Update app.py

app.py

@@ -41,19 +41,13 @@ def cache_pil_image(image: Image.Image) -> str:
 
 def upload(files: Optional[List[str]], chatbot: CHAT_HISTORY) -> CHAT_HISTORY:
     for file in files:
-        try:
-            # Intenta cargar y procesar imágenes
-            image = Image.open(file).convert('RGB')
-            image_preview = preprocess_image(image)
-            if image_preview:
-                # Mostrar una vista previa de la imagen cargada
-                gr.Image(image_preview).render()
-            image_path = cache_pil_image(image)
-            chatbot.append(((image_path,), None))
-        except Exception as e:
-            # Si no es una imagen, no mostramos vista previa y solo se guarda el archivo
-            print(f"Archivo no es una imagen, se sube como archivo: {file}")
-            chatbot.append(((file,), None))
+        image = Image.open(file).convert('RGB')
+        image_preview = preprocess_image(image)
+        if image_preview:
+            # Display a preview of the uploaded image
+            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):
@@ -64,7 +58,8 @@ def user(text_prompt: str, chatbot: CHAT_HISTORY):
 def bot(
     files: Optional[List[str]],
     model_choice: str,
-    chatbot: CHAT_HISTORY
+    chatbot: CHAT_HISTORY,
+    system_prompt: str  # Añadimos el parámetro para el system prompt
 ):
     if not GOOGLE_API_KEY:
         raise ValueError("GOOGLE_API_KEY is not set.")
@@ -79,9 +74,13 @@ def bot(
     )
 
     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 file.endswith(('jpg', 'jpeg', 'png'))] if files else []
+    image_prompt = [preprocess_image(Image.open(file).convert('RGB')) for file in files] if files else []
+    
+    # Aquí incluimos el system prompt en el flujo de generación de contenido
+    prompt = [system_prompt] + text_prompt  # El prompt del sistema se agrega al principio
+    
     model = genai.GenerativeModel(model_choice)
-    response = model.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
+    response = model.generate_content(prompt + image_prompt, stream=True, generation_config=generation_config)
 
     chatbot[-1][1] = ""
     for chunk in response:
@@ -101,7 +100,7 @@ text_prompt_component = gr.Textbox(
     placeholder="Message...", show_label=False, autofocus=True, scale=8
 )
 upload_button_component = gr.UploadButton(
-    label="Upload Files", file_count="multiple", file_types=["image", "pdf", "text"], scale=1
+    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(
@@ -110,6 +109,9 @@ model_choice_component = gr.Dropdown(
     label="Select Model",
     scale=2
 )
+system_prompt_component = gr.Textbox(
+    placeholder="Enter system prompt...", show_label=True, scale=8
+)
 
 user_inputs = [
     text_prompt_component,
@@ -119,6 +121,7 @@ user_inputs = [
 bot_inputs = [
     upload_button_component,
     model_choice_component,
+    system_prompt_component,  # Añadimos el prompt del sistema
     chatbot_component
 ]
 
@@ -132,6 +135,7 @@ with gr.Blocks() as demo:
             upload_button_component.render()
             run_button_component.render()
         model_choice_component.render()
+        system_prompt_component.render()  # Mostrar el campo para el system prompt
 
     run_button_component.click(
         fn=user,