Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import faiss
 import pickle
 import os
+import json
 from sentence_transformers import SentenceTransformer
 from exa_py import Exa
 from groq import Groq
@@ -15,7 +16,6 @@ app = FastAPI()
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 EXA_API_KEY = os.getenv("EXA_API_KEY")
 
-# clients
 client = Groq(api_key=GROQ_API_KEY)
 exa = Exa(api_key=EXA_API_KEY)
 
@@ -58,7 +58,7 @@ def ask_qwen(prompt):
     completion = client.chat.completions.create(
         model="qwen/qwen3-32b",
         messages=[
-            {"role": "system", "content": "You are a strict fact-checking analyst."},
+            {"role": "system", "content": "You are a strict fact-checking analyst. Return ONLY valid JSON."},
             {"role": "user", "content": prompt}
         ],
         temperature=0.3,
@@ -67,7 +67,7 @@ def ask_qwen(prompt):
     return completion.choices[0].message.content
 
 # =============================
-# 🧠 PIPELINE
+# 🧠 PIPELINE (FIXED)
 # =============================
 def analyze_problem(problem):
 
@@ -86,16 +86,31 @@ Problem:
 Evidence:
 {context}
 
-Output:
-Status: SOLVED or UNSOLVED
-Reason: one short sentence
-Gaps:
-- bullet points
-New Problem:
-rewrite
+Return ONLY valid JSON in this format:
+
+{{
+  "status": "SOLVED or UNSOLVED",
+  "reason": "one short sentence",
+  "gaps": ["gap1", "gap2"],
+  "new_problem": "rewrite of the problem"
+}}
 """
 
-    return ask_qwen(prompt)
+    response = ask_qwen(prompt)
+
+    try:
+        parsed = json.loads(response)
+    except:
+        parsed = {
+            "status": "ERROR",
+            "reason": "invalid JSON from model",
+            "gaps": [],
+            "new_problem": ""
+        }
+
+    parsed["problem"] = problem
+
+    return parsed
 
 # =============================
 # 🚀 API
@@ -111,7 +126,13 @@ async def analyze(file: UploadFile = File(...)):
         try:
             results.append(analyze_problem(p))
         except Exception as e:
-            results.append(f"ERROR: {str(e)}")
+            results.append({
+                "problem": p,
+                "status": "ERROR",
+                "reason": str(e),
+                "gaps": [],
+                "new_problem": ""
+            })
 
     return {
         "success": True,