Update app.py

app.py

@@ -267,6 +267,41 @@ def analyze_segment_with_gemini(cluster_text, is_full_text=False):
             }}
         """
 
+    response = llm.invoke(prompt)
+    
+    response_text = response.content
+    
+    try:
+        json_match = re.search(r'\{[\s\S]*\}', response_text)
+        if json_match:
+            response_json = json.loads(json_match.group(0))
+        else:
+            response_json = json.loads(response_text)
+        
+        return response_json
+    except json.JSONDecodeError as e:
+        print(f"Error parsing JSON response: {e}")
+        print(f"Raw response: {response_text}")
+        
+        if is_full_text:
+            return {
+                "segments": [
+                    {
+                        "topic_name": "JSON Parsing Error",
+                        "key_concepts": ["Error in response format"],
+                        "summary": f"Could not parse the API response. Raw text: {response_text[:200]}...",
+                        "quiz_questions": []
+                    }
+                ]
+            }
+        else:
+            return {
+                "topic_name": "JSON Parsing Error",
+                "key_concepts": ["Error in response format"],
+                "summary": f"Could not parse the API response. Raw text: {response_text[:200]}...",
+                "quiz_questions": []
+            }
+
 def process_document_with_quiz(text):
     token_count = len(tokenizer.encode(text))
     print(f"Text contains {token_count} tokens")