Update app.py

app.py

@@ -1,17 +1,16 @@
 import os
 import google.generativeai as genai
 import gradio as gr
-from PIL import Image
 
 # Configure the API
 genai.configure(api_key=os.environ["API_KEY"])
 
 # Create the model
 generation_config = {
-    "temperature": 1,
+    "temperature": 0.9,
     "top_p": 0.95,
     "top_k": 40,
-    "max_output_tokens": 8192,
+    "max_output_tokens": 1024,
 }
 
 model = genai.GenerativeModel(
@@ -20,43 +19,23 @@ model = genai.GenerativeModel(
 )
 
 # Initialize the chat session
-chat_session = model.start_chat(history=[])
-
-def process_file(file):
-    if file.name.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp')):
-        return Image.open(file.name)
-    elif file.name.lower().endswith(('.mp3', '.wav', '.ogg')):
-        return file.name  # Return the file path for audio
-    elif file.name.lower().endswith(('.mp4', '.avi', '.mov')):
-        return file.name  # Return the file path for video
-    else:
-        return None
-
-def respond(message, chat_history, files):
-    processed_files = []
-    if files:
-        processed_files = [process_file(file) for file in files if file is not None]
-    
-    if processed_files:
-        response = chat_session.send_message([message] + processed_files)
-    else:
-        response = chat_session.send_message(message)
-    
-    chat_history.append((message, response.text))
-    return "", chat_history
+chat = model.start_chat(history=[])
 
+def respond(message, history):
+    response = chat.send_message(message)
+    history.append((message, response.text))
+    return "", history
+
+# Create the Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Gemini Multimodal Chatbot")
-    gr.Markdown("Chat with the Gemini 1.5 Pro model. You can send text, images, audio, and video!")
+    gr.Markdown("# Simple Gemini Chatbot")
+    gr.Markdown("Chat with the Gemini 1.5 Pro model.")
     
     chatbot = gr.Chatbot()
-    msg = gr.Textbox()
+    msg = gr.Textbox(label="Your message")
     clear = gr.ClearButton([msg, chatbot])
-    
-    file_output = gr.File()
-    upload_button = gr.UploadButton("📁 Upload files", file_types=["image", "audio", "video"], file_count="multiple")
 
-    msg.submit(respond, [msg, chatbot, upload_button], [msg, chatbot])
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
 if __name__ == "__main__":
     demo.launch()