Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import google.generativeai as genai
 import gradio as gr
+from google.generativeai.protos import content_types as protos
 
 # Configure the API key
 genai.configure(api_key=os.environ["YOUR_API_KEY"])
@@ -25,8 +26,15 @@ model = genai.GenerativeModel(
 def chat_with_model(user_input, history):
     # Update the history with the user's input
     history.append({"role": "user", "content": user_input})
-    # Start or continue the chat session
-    chat_session = model.start_chat(history=history)
+    
+    # Convert history to expected format (list of protos.Content)
+    formatted_history = []
+    for entry in history:
+        content = protos.Content(text=entry["content"], role=entry["role"])
+        formatted_history.append(content)
+    
+    # Start or continue the chat session using the formatted history
+    chat_session = model.start_chat(history=formatted_history)
     response = chat_session.send_message(user_input)
     # Add the assistant's response to the history
     history.append({"role": "assistant", "content": response.text})
@@ -48,7 +56,7 @@ with gr.Blocks() as demo:
             show_label=False,
             placeholder="Type your message and press Enter"
         )
-        send_btn = gr.Button("Send")  # Ensure this line uses 'send_btn'
+        send_btn = gr.Button("Send")
     # Event handlers
     send_btn.click(chat_with_model, [user_input, state], [chatbot, state])
     user_input.submit(chat_with_model, [user_input, state], [chatbot, state])
@@ -56,4 +64,4 @@ with gr.Blocks() as demo:
     send_btn.click(lambda: "", None, user_input)
     user_input.submit(lambda: "", None, user_input)
 
-demo.launch()
+demo.launch()