Update app.py

app.py

@@ -54,32 +54,40 @@ def extract_text_from_pdf(pdf_file):
 def generate_image(description):
     try:
         client = genai.Client(api_key=GEMINI_API_KEY)
-        
+
         # Suponiendo que la API de Gemini tiene un endpoint para generar imágenes
-        model = "gemini-2.0-flash-thinking-exp-01-21"  # Cambiar según el modelo de imagen
+        model = "gemini-image-model"  # Suponiendo que este es el modelo correcto
         contents = [
             types.Content(
                 role="user",
                 parts=[types.Part.from_text(text=description)],
             ),
         ]
+        
+        # Configuración de la generación de imágenes, puedes ajustar estos parámetros
         generate_content_config = types.GenerateContentConfig(
             temperature=0.7,
             top_p=0.95,
             top_k=64,
-            max_output_tokens=65536,
-            response_mime_type="image/png",  # Suponiendo que la respuesta es en formato de imagen
+            max_output_tokens=2048,  # Reducir los tokens si es necesario
+            response_mime_type="image/png",  # Ajustar a "image/jpeg" si la API lo requiere
         )
         
+        # Suponiendo que la API devuelva una URL de la imagen
         response_image_url = ""
         for chunk in client.models.generate_content_stream(
             model=model,
             contents=contents,
             config=generate_content_config,
         ):
-            response_image_url += chunk.text  # Suponiendo que la API devuelve una URL de la imagen generada
+            response_image_url += chunk.text  # Se supone que aquí obtenemos la URL de la imagen generada
 
-        return response_image_url
+        # Verifica si obtuviste una URL
+        if response_image_url:
+            return response_image_url
+        else:
+            return "No se pudo generar la imagen, la respuesta está vacía."
+        
     except Exception as e:
         return f"Se produjo un error al generar la imagen: {e}"