Update app.py

app.py

@@ -41,11 +41,11 @@ except Exception as e:
     print(f"ERRO ao inicializar o cliente Groq: {e}.")
 
 
-# --- NOVA ROTA PARA LISTAR MODELOS DINAMICAMENTE ---
+# --- ROTA PARA LISTAR MODELOS DINAMICAMENTE ---
 @app.route('/list-models')
 def list_models():
     available_models = []
-    # CORREÇÃO: Lista de modelos Groq para geração de texto foi atualizada.
+    # Lista de modelos Groq para geração de texto foi atualizada.
     groq_text_models = [
         "llama-3.1-8b-instant", 
         "llama-3.3-70b-versatile", 
@@ -54,7 +54,6 @@ def list_models():
     ]
 
     try:
-        # Busca modelos Gemini
         if genai_client:
             for m in genai_client.list_models():
                 if 'generateContent' in m.supported_generation_methods:
@@ -65,7 +64,6 @@ def list_models():
                             "name": m.display_name
                         })
 
-        # Adiciona modelos Groq
         if groq_client:
             for model_id in groq_text_models:
                  display_name = model_id.split('/')[-1].replace('-instant', '').replace('-versatile', '')
@@ -76,7 +74,6 @@ def list_models():
                 
     except Exception as e:
         print(f"Erro ao listar modelos: {e}")
-        # Retorna uma lista de fallback em caso de erro na API
         return jsonify([
             {"value": "gemini:gemini-2.5-flash-latest", "name": "Gemini 2.5 Flash (Fallback)"},
             {"value": "groq:llama-3.1-8b-instant", "name": "Llama 3.1 8B (Fallback)"}
@@ -91,8 +88,7 @@ def list_models():
 def tts_proxy():
     data = request.get_json()
     text = data.get('text', '')
-    if not text:
-        return jsonify({"error": "No text provided"}), 400
+    if not text: return jsonify({"error": "No text provided"}), 400
     try:
         tts = gTTS(text=text, lang='en', tld='co.uk')
         mp3_fp = io.BytesIO()
@@ -125,7 +121,7 @@ def explain_proxy():
         
         if for_flashcard:
             schema = {"type": "object", "properties": {"term": {"type": "string"}, "translation": {"type": "string"}, "context_sentence": {"type": "string"}, "gapped_sentence": {"type": "string"}, "definition": {"type": "string"}}, "required": ["term", "translation", "context_sentence", "gapped_sentence", "definition"]}
-            prompt = f"Analyze '{word}' in context: '{context}'. Generate a JSON for a flashcard. The 'gapped_sentence' must replace '{word}' with '______________'. You must strictly follow the JSON schema and provide valid, non-empty values for all fields."
+            prompt = f"Analyze '{word}' in context: '{context}'. Generate a JSON for a flashcard. The 'gapped_sentence' must replace '{word}' with '______________'."
             return jsonify(get_ai_text_response(model_provider, model_name, system_instruction_base, prompt, json_schema=schema))
         
         else:
@@ -144,8 +140,7 @@ def activity_feedback():
     original_prompt = data.get('original_prompt', '')
     user_response = data.get('user_response', '')
 
-    if not original_prompt or not user_response:
-        return jsonify({"error": "Original prompt and user response are required."}), 400
+    if not original_prompt or not user_response: return jsonify({"error": "Original prompt and user response are required."}), 400
 
     if (model_provider == 'gemini' and not genai_client) or (model_provider == 'groq' and not groq_client):
         return jsonify({"error": f"{model_provider.upper()}_API_KEY not configured."}), 503
@@ -225,7 +220,6 @@ def generate_image():
     prompt = data.get('prompt')
     if not prompt: return jsonify({"error": "Image prompt is required."}), 400
     try:
-        # Usa um modelo de imagem específico e estável
         model = genai_client.GenerativeModel(model_name='gemini-2.5-flash-image-preview')
         response = model.generate_content(prompt)
         base64_image_data = response.parts[0].inline_data.data
@@ -236,19 +230,37 @@ def generate_image():
 # --- FUNÇÃO AUXILIAR E ROTA RAIZ ---
 def get_ai_text_response(provider, model_name, system_instruction, user_prompt, json_schema=None):
     if provider == 'gemini':
-        # Usa o model_name diretamente, que agora vem da lista dinâmica
         model = genai_client.GenerativeModel(model_name, system_instruction=system_instruction)
         config = {}
         if json_schema:
             config = {"response_mime_type": "application/json", "response_schema": json_schema}
         response = model.generate_content(user_prompt, generation_config=config)
-        return json.loads(response.text) if json_schema else response.text.strip()
         
+        parsed_json = json.loads(response.text)
+        if json_schema:
+            required_keys = json_schema.get("required", [])
+            if not all(key in parsed_json and parsed_json[key] for key in required_keys):
+                raise ValueError(f"AI response missing required keys. Required: {required_keys}, Got: {list(parsed_json.keys())}")
+        return parsed_json
+
     elif provider == 'groq':
-        messages = [{"role": "system", "content": system_instruction}, {"role": "user", "content": user_prompt}]
+        final_user_prompt = user_prompt
+        if json_schema:
+             final_user_prompt += f"\n\nYou MUST respond with a single JSON object that strictly follows this schema. Do not add any other text before or after the JSON object:\n{json.dumps(json_schema)}"
+        
+        messages = [{"role": "system", "content": system_instruction}, {"role": "user", "content": final_user_prompt}]
         config = {'response_format': {"type": "json_object"}} if json_schema else {}
         response = groq_client.chat.completions.create(model=model_name, messages=messages, **config)
-        return json.loads(response.choices[0].message.content) if json_schema else response.choices[0].message.content.strip()
+        
+        if json_schema:
+            parsed_json = json.loads(response.choices[0].message.content)
+            required_keys = json_schema.get("required", [])
+            if not all(key in parsed_json and parsed_json[key] for key in required_keys):
+                raise ValueError(f"AI response missing required keys. Required: {required_keys}, Got: {list(parsed_json.keys())}")
+            return parsed_json
+        else:
+            return response.choices[0].message.content.strip()
+
     raise Exception(f"Unsupported provider: {provider}")
 
 @app.route('/')