Update app.py

app.py

@@ -58,9 +58,12 @@ st.title("Gemini API Chatbot")
 # Get chat history from session state
 chat_history = st.session_state.get("chat_history", [])
 
+# Convert chat history to the expected format
+chat_history = [{"role": message["role"], "parts": [{"text": message["content"]}]} for message in chat_history]
+
 # Display previous messages
 for message in chat_history:
-   role, text = message["role"], message["content"]
+   role, text = message["role"], message["parts"][0]["text"]
    st.markdown(f"**{role.title()}:** {text}")
 
 # Get user input from text box
@@ -69,12 +72,12 @@ user_input = st.text_input("You")
 # Check if user input is not empty
 if user_input:
  # Add user message to chat history
- chat_history.append({"role": "user", "content": user_input})
+ chat_history.append({"role": "user", "parts": [{"text": user_input}]})
 
  # Display user message with markdown
  st.markdown(f"**You:** {user_input}")
 
- # Get model response with generate_content method
+ # Get model response with generate_content method```
  with st.spinner("Thinking..."):
      response = model.generate_content(chat_history)
 
@@ -82,7 +85,7 @@ if user_input:
  response_text = response.result.text
 
  # Add response message to chat history
- chat_history.append({"role": "assistant", "content": response_text})
+ chat_history.append({"role": "assistant", "parts": [{"text": response_text}]})
 
  # Display response message with markdown
  st.markdown(f"**Gemini Bot:** {response_text}")
@@ -108,7 +111,7 @@ if st.button("Clear History"):
 
 # Save chat history to database
 for message in chat_history:
-   c.execute("INSERT INTO history VALUES (?, ?)", (message["role"], message["content"]))
+   c.execute("INSERT INTO history VALUES (?, ?)", (message["role"], message["parts"][0]["text"]))
 conn.commit()
 
 # Close the connection