Update app.py

app.py

@@ -134,24 +134,18 @@ def pronunciation_feedback():
     except Exception as e:
         return jsonify({"error": f"Pronunciation analysis failed: {e}"}), 500
 
-# --- ROTA 6: GERAÇÃO DE IMAGEM (NOVO) ---
+# --- ROTA 6: GERAÇÃO DE IMAGEM ---
 @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
+    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
+    if not prompt: return jsonify({"error": "Image prompt is required."}), 400
     try:
-        # Usa o modelo "Nano Banana"
         model = genai_client.GenerativeModel(model_name='gemini-1.5-flash-image-preview')
         response = model.generate_content(prompt)
-        
-        # 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:
         return jsonify({"error": f"Image generation failed: {e}"}), 500
 
@@ -160,10 +154,18 @@ def get_ai_text_response(provider, model_name, system_instruction, user_prompt,
     if provider == 'gemini':
         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
-        generation_config = genai.types.GenerationConfig(**config) if config else None
+        
+        # CORREÇÃO: Constrói o objeto GenerationConfig corretamente, como esperado pela biblioteca.
+        generation_config = None
+        if json_schema:
+            generation_config = genai.types.GenerationConfig(
+                response_mime_type="application/json",
+                response_schema=json_schema
+            )
+        
         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}]
         config = {'response_format': {"type": "json_object"}} if json_schema else {}