Update main.py

main.py

@@ -663,25 +663,55 @@ def call_gemini(prompt_text, is_json_output_expected=True):
             {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
             {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
         ]
-        gen_config = genai.types.GenerationConfig(response_mime_type="application/json") if is_json_output_expected else None
-        response = model.generate_content(prompt_text, generation_config=gen_config, safety_settings=safety_settings)
+        
+        # Fixed: Use proper GenerationConfig parameters
+        gen_config = None
+        if is_json_output_expected:
+            gen_config = genai.types.GenerationConfig(
+                temperature=0.2,  # Lower temperature for more consistent JSON
+                top_p=0.8,
+                top_k=40,
+                max_output_tokens=2048,
+            )
+        
+        response = model.generate_content(
+            prompt_text, 
+            generation_config=gen_config, 
+            safety_settings=safety_settings
+        )
 
         if not response.parts:
             reason = "Unknown"
             try:
                 reason = response.prompt_feedback.block_reason.name if hasattr(response, 'prompt_feedback') and response.prompt_feedback.block_reason else None
-                if not reason and response.candidates: reason = response.candidates[0].finish_reason.name
-            except Exception: pass
+                if not reason and response.candidates: 
+                    reason = response.candidates[0].finish_reason.name
+            except Exception: 
+                pass
             return None, f"AI response empty or blocked. Reason: {reason}"
+        
         if not hasattr(response, 'text') or not response.text:
             return None, "AI response text empty."
 
         if is_json_output_expected:
-            parsed_json, err = safe_json_loads(response.text)
-            if err: return None, f"AI response received, but failed to parse as JSON: {err}. Raw: {response.text[:500]}..."
+            # Clean the response text to extract JSON
+            response_text = response.text.strip()
+            # Remove markdown code blocks if present
+            if response_text.startswith('```json'):
+                response_text = response_text[7:]
+            if response_text.startswith('```'):
+                response_text = response_text[3:]
+            if response_text.endswith('```'):
+                response_text = response_text[:-3]
+            response_text = response_text.strip()
+            
+            parsed_json, err = safe_json_loads(response_text)
+            if err: 
+                return None, f"AI response received, but failed to parse as JSON: {err}. Raw: {response_text[:500]}..."
             return parsed_json, None
         else:
-            return response.text, None
+            return response_text, None
+            
     except Exception as e:
         app.logger.error(f"Gemini API Call Error: {e} (Type: {type(e).__name__})")
         return None, f"Gemini API Call Error: {e}"
@@ -703,6 +733,8 @@ JSON Output:
 """
     return call_gemini(prompt, is_json_output_expected=True)
 
+
+
 def generate_quiz_questions_gemini(members, relationships, num_questions=3):
     if not members or len(members) < 2: return None, "Need at least 2 members for a quiz."
     id_to_name = {p["id"]: p.get("name", "Unknown") for p in members if isinstance(p, dict)}
@@ -1498,24 +1530,42 @@ def get_proposal_diff_api(proposer_phone_to_review):
 
 # --- AI Feature Endpoints (Protected as they operate on user's tree or consume user quota) ---
 @app.route('/ai/build_tree_from_text', methods=['POST'])
-@token_required # AI features should be protected
+@token_required
 def ai_build_tree_api():
-    # This endpoint doesn't directly modify a tree, just provides analysis.
-    # The owner_phone_of_tree context isn't strictly needed here unless we log usage against a tree.
-    # For now, it's a general utility for an authenticated user.
-    if api_key_error: return jsonify({"error": "Gemini API not configured"}), 503
+    """
+    Endpoint to analyze family description text and extract people and relationships.
+    
+    Request body:
+    {
+        "description": "Family description text here..."
+    }
+    
+    Response:
+    {
+        "people": [...],
+        "relationships": [...]
+    }
+    """
+    if api_key_error: 
+        return jsonify({"error": "Gemini API not configured"}), 503
 
     req_data = request.json
     if not req_data or "description" not in req_data:
         return jsonify({"error": "Missing description in request body"}), 400
+    
     description = req_data["description"]
+    
+    if not description.strip():
+        return jsonify({"error": "Description cannot be empty"}), 400
 
     ai_result, error = generate_tree_from_description_gemini(description)
     if error:
         return jsonify({"error": "AI analysis failed", "detail": error}), 500
+    
     return jsonify(ai_result), 200
 
 
+
 @app.route('/tree/<string:owner_phone_of_tree>/ai_merge_suggestions', methods=['POST'])
 @token_required
 @phone_required