Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
-from google.generativeai import GenerativeModel, Part, Content
+from google.generativeai import GenerativeModel
 
 # Configure API key
-api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNME"
+api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
 model_name = "gemini-pro"
 
 # Create model object
@@ -19,11 +19,8 @@ user_input = st.text_input("You")
 
 # Check if user input is not empty
 if user_input:
-    # Create user message object
-    user_message = Content(parts=[Part(text=user_input)], role="user")
-
     # Add user message to chat history
-    chat_history.append(user_message)
+    chat_history.append(user_input)
 
     # Display user message with markdown
     st.markdown(f"**You:** {user_input}")
@@ -36,7 +33,7 @@ if user_input:
     response_text = response.contents[-1].parts[0].text
 
     # Add response message to chat history
-    chat_history.append(Content(parts=[Part(text=response_text)], role="assistant"))
+    chat_history.append(response_text)
 
     # Display response message with markdown
     st.markdown(f"**Gemini Bot:** {response_text}")