Update app/services/extraction.py

app/services/extraction.py

@@ -362,12 +362,14 @@ Output:
     model = genai.GenerativeModel('gemini-2.5-flash')
     
     # Generate response
-    response = model.generate_content(prompt, 
-                                     generation_config=genai.types.GenerationConfig(
-                                         temperature=0,
-                                         max_output_tokens=150,
-                                         top_p=1
-                                     ))
+    response = model.generate_content(
+        prompt, 
+        generation_config=genai.types.GenerationConfig(
+            temperature=0,
+            max_output_tokens=500,  # Safety net, not a hard limit
+            top_p=1,
+            response_mime_type="application/json"  # 🔥 FORCES JSON OUTPUT
+        ))
     
     # Process the response
     output_text = response.text.strip()