Reupload OmniDev clean version

app/api/augment/route.ts

@@ -65,7 +65,20 @@ export async function POST(req: NextRequest) {
         ],
         config: { maxOutputTokens: 4096 },
       } as any);
-      text = (res as any)?.candidates?.[0]?.content?.parts?.[0]?.text || "";
+      text = ((res as any)?.response?.text && (res as any).response.text())
+        || (res as any)?.text
+        || ((res as any)?.candidates?.[0]?.content?.parts?.map((p: any) => p?.text || "").join("") || "");
+      // Fallback: single user message with combined system + prompt
+      if (!text || !String(text).trim()) {
+        const res2 = await ai.models.generateContent({
+          model: model || "gemini-2.5-flash",
+          contents: [ { role: 'user', parts: [{ text: `${SYS}\n\n${userPrompt}` }] } ],
+          config: { maxOutputTokens: 4096 },
+        } as any);
+        text = ((res2 as any)?.response?.text && (res2 as any).response.text())
+          || (res2 as any)?.text
+          || ((res2 as any)?.candidates?.[0]?.content?.parts?.map((p: any) => p?.text || "").join("") || "");
+      }
     } else {
       const token = process.env.HF_TOKEN || process.env.DEFAULT_HF_TOKEN;
       const client = new InferenceClient(token);
@@ -101,4 +114,3 @@ export async function POST(req: NextRequest) {
     return NextResponse.json({ ok: false, message: e?.message || "Internal error" } as AugmentResponse, { status: 500 });
   }
 }
-

app/api/enhance/route.ts

@@ -35,7 +35,19 @@ export async function POST(req: NextRequest) {
         ],
         config: { maxOutputTokens: 1024 },
       } as any);
-      text = (res as any)?.candidates?.[0]?.content?.parts?.[0]?.text || '';
+      text = ((res as any)?.response?.text && (res as any).response.text())
+        || (res as any)?.text
+        || ((res as any)?.candidates?.[0]?.content?.parts?.map((p: any) => p?.text || "").join("") || "");
+      if (!text || !String(text).trim()) {
+        const res2 = await ai.models.generateContent({
+          model: model || 'gemini-2.5-flash',
+          contents: [ { role: 'user', parts: [{ text: `${system}\n\n${userParts.join('\n')}` }] } ],
+          config: { maxOutputTokens: 1024 },
+        } as any);
+        text = ((res2 as any)?.response?.text && (res2 as any).response.text())
+          || (res2 as any)?.text
+          || ((res2 as any)?.candidates?.[0]?.content?.parts?.map((p: any) => p?.text || "").join("") || "");
+      }
     } else {
       const token = process.env.HF_TOKEN || process.env.DEFAULT_HF_TOKEN;
       const client = new InferenceClient(token);
@@ -58,4 +70,3 @@ export async function POST(req: NextRequest) {
     return NextResponse.json({ ok: false, message: e?.message || 'Internal error' }, { status: 500 });
   }
 }
-