Update app.py

app.py

@@ -60,7 +60,7 @@ chat_history = st.session_state.get("chat_history", [])
 
 # Display previous messages
 for message in chat_history:
-   role, text = message
+   role, text = message["role"], message["content"]
    st.markdown(f"**{role.title()}:** {text}")
 
 # Get user input from text box
@@ -69,7 +69,7 @@ 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", "parts": [user_input]})
+ chat_history.append({"role": "user", "content": user_input})
 
  # Display user message with markdown
  st.markdown(f"**You:** {user_input}")
@@ -79,10 +79,10 @@ if user_input:
      response = model.generate_content(chat_history)
 
  # Get response text from response object
- response_text = response.text
+ response_text = response.result.text
 
  # Add response message to chat history
- chat_history.append({"role": "assistant", "parts": [response_text]})
+ chat_history.append({"role": "assistant", "content": response_text})
 
  # Display response message with markdown
  st.markdown(f"**Gemini Bot:** {response_text}")
@@ -108,7 +108,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["parts"][0]))
+   c.execute("INSERT INTO history VALUES (?, ?)", (message["role"], message["content"]))
 conn.commit()
 
 # Close the connection