[BACKEND - FIXED GEMINI ERROR] | fixed the gemini error and adjusted the controllers

Author: AngadSudan

.husky/pre-commit

@@ -0,0 +1 @@
+bun format

apps/client/src/components/Project/Editor.tsx

@@ -23,35 +23,28 @@ export default function Editor({
 
     if (Object.keys(files).length > 0) {
       const timeout = setTimeout(() => {
-        console.log(githubUrl);
-        console.log(githubUrl.replace("https://api.github.com/repos/", ""));
-        console.log(
-          githubUrl
-            .replace("https://api.github.com/repos/", "")
-            .replace(".git", "")
-        );
-        sdk
-          .embedGithubProject(
-            containerRef.current!,
-            githubUrl
-              .replace("https://api.github.com/repos/", "")
-              .replace(".git", "")
-          )
-          .then((vm) => {
-            if (!cancelled) {
-              setIsLoaded(true);
-              vmRef.current = vm;
-            }
-          })
-          .catch((err) => {
-            console.error("StackBlitz embed error:", err);
-
-            // Set error state
-            setError(err?.message || "Failed to load StackBlitz");
+        // sdk
+        //   .embedGithubProject(
+        //     containerRef.current!,
+        //     githubUrl
+        //       .replace("https://api.github.com/repos/", "")
+        //       .replace(".git", "")
+        //   )
+        //   .then((vm) => {
+        //     if (!cancelled) {
+        //       setIsLoaded(true);
+        //       vmRef.current = vm;
+        //     }
+        //   })
+        //   .catch((err) => {
+        //     console.error("StackBlitz embed error:", err);
 
+        //     // Set error state
+        //     setError(err?.message || "Failed to load StackBlitz");
+        //   })
             // Optional fallback: embed a custom project if GitHub embed fails
             // Uncomment if you want fallback
-            /*
+            
       sdk.embedProject(
         containerRef.current!,
         {
@@ -77,9 +70,9 @@ export default function Editor({
         console.error("Fallback embed error:", fallbackErr);
         setError(fallbackErr?.message || "Failed to load StackBlitz fallback");
       });
-      */
-          });
-      }, 300);
+      
+          // });
+      }, 1000);
 
       return () => {
         cancelled = true;

apps/server/controller/AiFeatures.controllers.ts

@@ -6,6 +6,7 @@ import {
   generateFileStructurePrompt,
   generateSDDDocument,
 } from "../utils/prompt";
+import {parse} from 'yaml';
 
 class GeminiAiFeatures {
   availableKey: string;
@@ -41,23 +42,20 @@ class GeminiAiFeatures {
     const response = await this.jsonModel.models.generateContent({
       model: "gemini-2.0-flash",
       contents: prompt,
-      config: {
-        responseMimeType: "application/json",
-        responseSchema: {
-          type: Type.ARRAY,
-          items: {
-            type: Type.OBJECT,
-            properties: {
-              path: { type: Type.STRING },
-              content: { type: Type.STRING },
-            },
-            required: ["path", "content"],
-          },
-        },
-      },
     });
 
-    return response.text;
+    console.log(response.text);
+    console.log("_______________________");
+    const parsedText = response.text.replace('```yaml',"").replace("```","")
+    // TODO: use a json to yaml converter inorder to get the repsonse is json and return the project_dir object value
+    // inside of an array
+    const parsedObject = parse(parsedText);
+    console.log(parsedObject);
+    console.log("_______________________");
+    
+    console.log(parsedObject.project_dir);
+    
+    return parsedObject.project_dir;
   }
 
   async getResponseText(

apps/server/controller/project.controllers.ts

@@ -21,32 +21,30 @@ class ProjectController {
 
       const githubToken =
         dbUser.githubToken || "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";
-      const createRepository = await new githubController(
-        githubToken
-      ).createRepository(name, description);
-
-      if (!createRepository) {
-        throw new Error("Error in creating repository");
-      }
-
+      
       const generatedPrompt =
         await gptFeaturesControllers.enhanceUserGivenDescription(description);
+
+      console.log("generated prompt is : ",generatedPrompt);
+      
       if (!generatedPrompt) {
         throw new Error("Couln't generate SRS");
       }
 
       const generatedFileStructure =
-        await gptFeaturesControllers.generateProjectFolderStructure(
+        await AiFeaturesControllers.generateProjectFileStructure(
           generatedPrompt
         );
+
+      console.log("generated prompt is : ",generatedPrompt);
       if (!generatedFileStructure) {
         throw new Error("error in creating folder structure for your project");
       }
 
-      let createdFileStructure = JSON.parse(generatedFileStructure);
+      let createdFileStructure = generatedFileStructure;
       const createFileSystem: Record<string, string> = {};
 
-      if (Object.keys(createdFileStructure).length > 0) {
+      if (createdFileStructure.length > 0) {
         for (const file in createdFileStructure) {
           if (
             createdFileStructure[file].path &&
@@ -65,12 +63,21 @@ class ProjectController {
           }
         }
       }
-
+      console.log(createdFileStructure);
+      
       const getSDD = await AiFeaturesControllers.generateSDD(
         generatedPrompt,
         JSON.stringify(createFileSystem)
       );
 
+      const createRepository = await new githubController(
+        githubToken
+      ).createRepository(name, description);
+
+      if (!createRepository) {
+        throw new Error("Error in creating repository");
+      }
+
       const createdProject = await prisma.project.create({
         data: {
           name,

apps/server/package.json

@@ -25,6 +25,7 @@
     "@google/genai": "^1.9.0",
     "@google/generative-ai": "^0.24.1",
     "@octodock/octolog": "workspace:*",
+    "@octodock/prisma": "*",
     "@types/jsonwebtoken": "^9.0.10",
     "@types/passport": "^1.0.17",
     "apollo-server-express": "^3.13.0",
@@ -54,7 +55,7 @@
     "passport-github2": "^0.1.12",
     "prisma": "^6.11.0",
     "tsup": "^8.5.0",
-    "zod": "^4.0.5",
-    "@octodock/prisma": "*"
+    "yaml": "^2.8.1",
+    "zod": "^4.0.5"
   }
 }

apps/server/utils/keymanager.ts

@@ -71,10 +71,10 @@ class GeminiKeyManager {
 
 // Initialize the GeminiKeyManager with API keys
 const keyManager = new GeminiKeyManager([
+  "AIzaSyBninrss6yavN9W4amrNB9Fj4mfPZZZ-3Y",
   "AIzaSyA7c-oXClHSPlb26klR7H66KxjV7J_dobw",
-  "AIzaSyCWiWcnKPRQo0m-O7RXa1HRQpjMOx3RXC4",
   "AIzaSyC5N_RPX7j3TnTZpOpqeWt12zyxjGnPjIU",
-  "AIzaSyBninrss6yavN9W4amrNB9Fj4mfPZZZ-3Y"
+  "AIzaSyCWiWcnKPRQo0m-O7RXa1HRQpjMOx3RXC4",
 ]);
 
 export default keyManager;

apps/server/utils/prompt.ts

@@ -88,21 +88,20 @@ You are given a Software Requirements Specification (SRS) document.
 
 Your task is to return a realistic and production-grade backend project file structure that implements the described system, using best practices for a Node.js or python or springboot backend project.
 
-### Output Format
+### Output Format YAML
 
 Respond with **only** a JavaScript or django or springboot array of file paths, like this:
 
-[
-  'src/index.js',
-  'src/database.js',
-  'src/routes.js',
-  'src/controller.js',
-  'src/model.js',
-]
+project_dir
+  - 'src/index.js',
+  - 'src/database.js',
+  - 'src/routes.js',
+  - 'src/controller.js',
+  - 'src/model.js',
 
 ### Rules
-
-- The array should contain full relative paths to each file.
+- The yaml should contain full relative paths to each file.
+- follow the strict yaml format and nothing more
 - Include meaningful folders and subfolders (e.g. src, routes, controllers, models, middleware, utils).
 - Follow best practices for organizing a scalable Node.js or django backend.
 - Do **not** include folder names as objects — just file paths as strings.

bun.lock

@@ -6,9 +6,11 @@
     "dev": "turbo run dev",
     "lint": "turbo run lint",
     "format": "prettier --write \"**/*.{ts,tsx,md}\"",
-    "check-types": "turbo run check-types"
+    "check-types": "turbo run check-types",
+    "prepare": "husky"
   },
   "devDependencies": {
+    "husky": "^9.1.7",
     "prettier": "^3.6.0",
     "turbo": "^2.5.4",
     "typescript": "5.8.2"