Update app.py

app.py

@@ -72,32 +72,21 @@ def explain_proxy():
     
     system_instruction_base = f"You are a professional English tutor focused on the context '{context_focus}'."
     try:
+        # Se for uma requisição de Geração de Atividade (custom_prompt)
         if custom_prompt:
-            return jsonify({"explanation": get_ai_text_response(model_provider, model_name, system_instruction_base, custom_prompt)})
+             # A resposta será texto simples, então retornamos em um campo JSON
+            activity_text = get_ai_text_response(model_provider, model_name, system_instruction_base, custom_prompt)
+            return jsonify({"explanation": activity_text}) # Reutilizando a chave 'explanation'
+        
         if not word: return jsonify({"error": "No word selected."}), 400
+        
+        # Lógica para Flashcard
         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 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)
+            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))
+        
+        # Lógica para tradução rápida
         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)
@@ -171,25 +160,18 @@ 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':
-        # 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)
-        
         config = {}
         if json_schema:
-            config = {
-                "response_mime_type": "application/json",
-                "response_schema": 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
+        return json.loads(response.text) if json_schema else response.text.strip()
         
     elif provider == 'groq':
         messages = [{"role": "system", "content": system_instruction}, {"role": "user", "content": 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
+        return json.loads(response.choices[0].message.content) if json_schema else response.choices[0].message.content.strip()
     raise Exception(f"Unsupported provider: {provider}")
 
 @app.route('/')