Update app.py

app.py

@@ -23,39 +23,15 @@ GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 if not GOOGLE_API_KEY:
     raise ValueError("GOOGLE_API_KEY is not set in environment variables.")
 
-IMAGE_CACHE_DIRECTORY = "/tmp"
 IMAGE_WIDTH = 512
 CHAT_HISTORY = List[Tuple[Optional[Union[Tuple[str], str]], Optional[str]]]
 
-def preprocess_image(image: Image.Image) -> Optional[Image.Image]:
-    if image:
-        image_height = int(image.height * IMAGE_WIDTH / image.width)
-        return image.resize((IMAGE_WIDTH, image_height))
-
-def cache_pil_image(image: Image.Image) -> str:
-    image_filename = f"{uuid.uuid4()}.jpeg"
-    os.makedirs(IMAGE_CACHE_DIRECTORY, exist_ok=True)
-    image_path = os.path.join(IMAGE_CACHE_DIRECTORY, image_filename)
-    image.save(image_path, "JPEG")
-    return image_path
-
-def upload(files: Optional[List[str]], chatbot: CHAT_HISTORY) -> CHAT_HISTORY:
-    for file in files:
-        image = Image.open(file).convert('RGB')
-        image_preview = preprocess_image(image)
-        if image_preview:
-            gr.Image(image_preview).render()
-        image_path = cache_pil_image(image)
-        chatbot.append(((image_path,), None))
-    return chatbot
-
 def user(text_prompt: str, chatbot: CHAT_HISTORY):
     if text_prompt:
         chatbot.append((text_prompt, None))
     return "", chatbot
 
 def bot(
-    files: Optional[List[str]],
     model_choice: str,
     system_instruction: Optional[str],  # Sistema de instrucciones opcional
     chatbot: CHAT_HISTORY
@@ -76,15 +52,14 @@ def bot(
         system_instruction = "1"  # O puedes poner un valor predeterminado como "No system instruction provided."
 
     text_prompt = [chatbot[-1][0]] if chatbot and chatbot[-1][0] and isinstance(chatbot[-1][0], str) else []
-    image_prompt = [preprocess_image(Image.open(file).convert('RGB')) for file in files] if files else []
     
     model = genai.GenerativeModel(
         model_name=model_choice,
         generation_config=generation_config,
-        system_instruction=system_instruction  # Usar el valor por defecto si está vacío
+        system_instruction=system_instruction
     )
 
-    response = model.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
+    response = model.generate_content(text_prompt, stream=True, generation_config=generation_config)
 
     chatbot[-1][1] = ""
     for chunk in response:
@@ -104,7 +79,6 @@ system_instruction_component = gr.Textbox(
 # Definir los componentes de entrada y salida
 chatbot_component = gr.Chatbot(label='Gemini', bubble_full_width=False, scale=2, height=300)
 text_prompt_component = gr.Textbox(placeholder="Message...", show_label=False, autofocus=True, scale=8)
-upload_button_component = gr.UploadButton(label="Upload Images", file_count="multiple", file_types=["image"], scale=1)
 run_button_component = gr.Button(value="Run", variant="primary", scale=1)
 model_choice_component = gr.Dropdown(
     choices=["gemini-1.5-flash", "gemini-2.0-flash-exp", "gemini-1.5-pro"],
@@ -114,7 +88,7 @@ model_choice_component = gr.Dropdown(
 )
 
 user_inputs = [text_prompt_component, chatbot_component]
-bot_inputs = [upload_button_component, model_choice_component, system_instruction_component, chatbot_component]
+bot_inputs = [model_choice_component, system_instruction_component, chatbot_component]
 
 # Definir la interfaz de usuario
 with gr.Blocks() as demo:
@@ -126,7 +100,6 @@ with gr.Blocks() as demo:
         chatbot_component.render()
         with gr.Row():
             text_prompt_component.render()
-            upload_button_component.render()
             run_button_component.render()
 
         # Crear el acordeón para la instrucción del sistema al final
@@ -151,12 +124,5 @@ with gr.Blocks() as demo:
         fn=bot, inputs=bot_inputs, outputs=[chatbot_component],
     )
 
-    upload_button_component.upload(
-        fn=upload,
-        inputs=[upload_button_component, chatbot_component],
-        outputs=[chatbot_component],
-        queue=False
-    )
-
 # Lanzar la aplicación
 demo.queue(max_size=99).launch(debug=False, show_error=True)