Update app.py

app.py

@@ -9,8 +9,6 @@ if not GOOGLE_API_KEY:
     raise ValueError("La variable de entorno GOOGLE_API_KEY no está definida.")
 
 genai.configure(api_key=GOOGLE_API_KEY)
-model_text = genai.GenerativeModel('gemini-pro')
-model_vision = genai.GenerativeModel('gemini-pro-vision')
 
 def read_text_file(filepath):
     with open(filepath, 'r', encoding='utf-8') as f:
@@ -39,15 +37,22 @@ def process_input(instruction, text_file, pdf_file, image_file):
         prompt_parts.append(f"Contenido del archivo PDF:\n{file_content}")
 
     if image_file:
-        # Para el modelo vision, necesitas pasar la imagen como datos
-        prompt_parts.append(image_file)
-        response = model_vision.generate_content(prompt_parts)
-        return response.text
+        # Aquí debes trabajar con la API de imágenes
+        image_data = image_file  # Usamos directamente la imagen cargada
+        response = genai.ImageGeneration.create(
+            model="gemini-1.5-flash",  # Usamos siempre el modelo correcto
+            image=image_data
+        )
+        return response['image']  # Asumiendo que la respuesta devuelve una imagen procesada
 
     if prompt_parts:
         prompt = "\n\n".join(prompt_parts)
-        response = model_text.generate_content(prompt)
-        return response.text
+        response = genai.ChatCompletion.create(
+            model="gemini-1.5-flash",  # Usamos siempre el modelo correcto
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=8192  # Cambié a 8192 tokens
+        )
+        return response['choices'][0]['message']['content']
     else:
         return "Por favor, proporciona alguna instrucción o archivo."
 
@@ -57,11 +62,11 @@ iface = gr.Interface(
         gr.Textbox(label="Instrucciones"),
         gr.File(label="Archivo de Texto (.txt)"),
         gr.File(label="Archivo PDF (.pdf)"),
-        gr.Image(label="Imagen"),
+        gr.Image(label="Imagen", type="file"),  # Cambié el tipo de 'image' a 'file'
     ],
     outputs=gr.Textbox(label="Respuesta de Gemini"),
     title="Interactúa con Gemini",
     description="Sube archivos de texto, PDF o imágenes y proporciona instrucciones para que Gemini los procese.",
 )
 
-iface.launch()
+iface.launch()