Update app.py

app.py

@@ -76,9 +76,28 @@ def explain_proxy():
             return jsonify({"explanation": get_ai_text_response(model_provider, model_name, system_instruction_base, custom_prompt)})
         if not word: return jsonify({"error": "No word selected."}), 400
         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 '______________'."
-            return jsonify(get_ai_text_response(model_provider, model_name, system_instruction_base, prompt, json_schema=schema))
+            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 the term '{word}' within the context: '{context}'. Generate a JSON object for a flashcard. The 'gapped_sentence' must replace the term '{word}' with '______________'."
+            
+            card_data = get_ai_text_response(model_provider, model_name, system_instruction_base, prompt, json_schema=schema)
+            
+            # Validação para garantir que a IA retornou os dados corretos
+            required_keys = schema['required']
+            if not all(key in card_data and card_data[key] for key in required_keys):
+                print(f"Dados incompletos da IA: {card_data}")
+                raise Exception("AI returned incomplete or invalid data for the flashcard.")
+
+            return jsonify(card_data)
         else:
             prompt = f"Analyze '{word}' in context: '{context}'. Provide a one-sentence English explanation, then '---', then the Portuguese translation."
             parts = get_ai_text_response(model_provider, model_name, system_instruction_base, prompt).split('---', 1)
@@ -152,11 +171,10 @@ 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':
-        full_model_name = f'models/{model_name}'
-        model = genai_client.GenerativeModel(full_model_name, system_instruction=system_instruction)
+        # CORREÇÃO: Removemos o prefixo 'models/' que estava causando o erro 404.
+        # A própria biblioteca gerencia isso.
+        model = genai_client.GenerativeModel(model_name, system_instruction=system_instruction)
         
-        # CORREÇÃO DEFINITIVA: O `generation_config` deve ser um dicionário simples.
-        # O uso do `genai.types.GenerationConfig` estava causando o erro de inicialização.
         config = {}
         if json_schema:
             config = {