Update app.py

app.py

@@ -49,20 +49,18 @@ def tts_proxy():
     if not text:
         return jsonify({"error": "No text provided"}), 400
     try:
-        tts = gTTS(text=text, lang='en', tld='co.uk') # Sotaque britânico para variedade
+        tts = gTTS(text=text, lang='en', tld='co.uk')
         mp3_fp = io.BytesIO()
         tts.write_to_fp(mp3_fp)
         mp3_fp.seek(0)
         return send_file(mp3_fp, mimetype='audio/mpeg', as_attachment=True, download_name='audio.mp3')
     except Exception as e:
-        print(f"Erro no gTTS: {e}")
         return jsonify({"error": f"Failed to generate audio via gTTS: {e}"}), 500
 
 
 # --- ROTA 2: ANÁLISE DE TEXTO (FLASHCARDS, ETC.) ---
 @app.route('/explain-proxy', methods=['POST'])
 def explain_proxy():
-    # ... (código existente sem alterações) ...
     data = request.get_json()
     model_provider, model_name = data.get('model', 'gemini:gemini-1.5-flash-latest').split(':', 1)
     word, context = data.get('word', '').strip(), data.get('context', '')
@@ -91,9 +89,9 @@ def explain_proxy():
 # --- ROTA 3: ANÁLISE DE IMAGEM ---
 @app.route('/analyze-image', methods=['POST'])
 def analyze_image():
-    # ... (código existente sem alterações) ...
     if not genai_client: return jsonify({"error": "GEMINI_API_KEY not configured."}), 503
-    data, base64_image = request.get_json(), data.get('image')
+    data = request.get_json()
+    base64_image = data.get('image')
     if not base64_image: return jsonify({"error": "No image data."}), 400
     try:
         image = Image.open(io.BytesIO(base64.b64decode(base64_image.split(',')[1])))
@@ -108,7 +106,6 @@ def analyze_image():
 # --- ROTA 4: CHAT COM IA ---
 @app.route('/chat-with-ai', methods=['POST'])
 def chat_with_ai():
-    # ... (código existente sem alterações) ...
     if not groq_client: return jsonify({"error": "GROQ_API_KEY not configured."}), 503
     data = request.get_json()
     history, user_message = data.get('history', []), data.get('message', '')
@@ -121,49 +118,51 @@ def chat_with_ai():
     except Exception as e:
         return jsonify({"error": f"AI chat failed: {e}"}), 500
 
-# --- ROTA 5: ANÁLISE DE PRONÚNCIA (NOVO) ---
+# --- ROTA 5: ANÁLISE DE PRONÚNCIA ---
 @app.route('/pronunciation-feedback', methods=['POST'])
 def pronunciation_feedback():
-    if not groq_client:
-        return jsonify({"error": "GROQ_API_KEY not configured for pronunciation analysis."}), 503
-    
+    if not groq_client: return jsonify({"error": "GROQ_API_KEY not configured for pronunciation analysis."}), 503
     data = request.get_json()
-    target_text = data.get('target_text')
-    user_text = data.get('user_text') # Texto transcrito do áudio do usuário
+    target_text, user_text = data.get('target_text'), data.get('user_text')
+    if not target_text or not user_text: return jsonify({"error": "Target and user text are required."}), 400
+    try:
+        system_instruction = "You are an expert American English pronunciation coach. The user tried to say a target sentence, and their speech was transcribed. Based on the likely pronunciation differences, provide brief, friendly, and actionable feedback in Portuguese. Focus on 1-2 key points. If it's very close, praise the user."
+        user_prompt = f"Target: \"{target_text}\"\nTranscription: \"{user_text}\"\n\nProvide pronunciation feedback."
+        messages = [{"role": "system", "content": system_instruction}, {"role": "user", "content": user_prompt}]
+        response = groq_client.chat.completions.create(model="llama-3.1-8b-instant", messages=messages, temperature=0.5)
+        return jsonify({"feedback": response.choices[0].message.content.strip()})
+    except Exception as e:
+        return jsonify({"error": f"Pronunciation analysis failed: {e}"}), 500
 
-    if not target_text or not user_text:
-        return jsonify({"error": "Target and user text are required."}), 400
-    
+# --- ROTA 6: GERAÇÃO DE IMAGEM (NOVO) ---
+@app.route('/generate-image', methods=['POST'])
+def generate_image():
+    if not genai_client:
+        return jsonify({"error": "GEMINI_API_KEY not configured for image generation."}), 503
+    data = request.get_json()
+    prompt = data.get('prompt')
+    if not prompt:
+        return jsonify({"error": "Image prompt is required."}), 400
     try:
-        system_instruction = "You are an expert American English pronunciation coach. The user tried to say a target sentence, and their speech was transcribed. Based on the likely pronunciation differences between the target and the transcription, provide brief, friendly, and actionable feedback in Portuguese. Focus on 1-2 key points. If the transcription is very close to the target, praise the user."
-        
-        user_prompt = f"Frase-alvo: \"{target_text}\"\nTranscrição do usuário: \"{user_text}\"\n\nPor favor, forneça o feedback de pronúncia."
-        
-        messages = [
-            {"role": "system", "content": system_instruction},
-            {"role": "user", "content": user_prompt}
-        ]
+        # Usa o modelo "Nano Banana"
+        model = genai_client.GenerativeModel(model_name='gemini-1.5-flash-image-preview')
+        response = model.generate_content(prompt)
         
-        response = groq_client.chat.completions.create(
-            model="llama-3.1-8b-instant", # Modelo rápido para feedback instantâneo
-            messages=messages,
-            temperature=0.5
-        )
-        
-        feedback = response.choices[0].message.content.strip()
-        return jsonify({"feedback": feedback})
+        # A API retorna a imagem em base64 diretamente nos 'parts'
+        base64_image_data = response.parts[0].inline_data.data
+        return jsonify({"image_base64": base64_image_data})
         
     except Exception as e:
-        print(f"Erro na análise de pronúncia: {e}")
-        return jsonify({"error": f"Pronunciation analysis failed: {e}"}), 500
+        return jsonify({"error": f"Image generation failed: {e}"}), 500
 
 # --- FUNÇÃO AUXILIAR E ROTA RAIZ ---
 def get_ai_text_response(provider, model_name, system_instruction, user_prompt, json_schema=None):
-    # ... (código existente sem alterações) ...
     if provider == 'gemini':
-        model = genai_client.GenerativeModel(model_name)
+        full_model_name = f'models/{model_name}'
+        model = genai_client.GenerativeModel(full_model_name, system_instruction=system_instruction)
         config = {"response_mime_type": "application/json", "response_schema": json_schema} if json_schema else None
-        response = model.generate_content(user_prompt, generation_config=config)
+        generation_config = genai.types.GenerationConfig(**config) if config else None
+        response = model.generate_content(user_prompt, generation_config=generation_config)
         return json.loads(response.text) if json_schema else response.text
     elif provider == 'groq':
         messages = [{"role": "system", "content": system_instruction}, {"role": "user", "content": user_prompt}]
@@ -177,4 +176,5 @@ def root():
     return send_file('index.html')
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', port=7860)
+