Update app.py

app.py

@@ -8,21 +8,21 @@ genai.configure(api_key="AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM")
 generation_config = {
   "temperature": 0.9,
   "top_p": 1,
-  "top_k": 1,
+  "top_k": 1,  
   "max_output_tokens": 2048   
 }
 
 safety_settings = [
   {
      "category": "HARM_CATEGORY_HARASSMENT",
-     "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+     "threshold": "BLOCK_MEDIUM_AND_ABOVE"  
   },
-  {
+  { 
      "category": "HARM_CATEGORY_HATE_SPEECH",
      "threshold": "BLOCK_MEDIUM_AND_ABOVE"
   },
   {
-     "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+     "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", 
      "threshold": "BLOCK_MEDIUM_AND_ABOVE"
   },
   {
@@ -32,46 +32,49 @@ safety_settings = [
 ]
 
 model = genai.GenerativeModel(
-  model_name="gemini-pro", 
-  generation_config=generation_config,
-  safety_settings=safety_settings
+   model_name="gemini-pro",
+   generation_config=generation_config, 
+   safety_settings=safety_settings  
 )
 
 # Chatbot interface
-st.title("Gemini API Chatbot")  
+st.title("Gemini API Chatbot")   
 
-chat_history = st.session_state.get("chat_history", []) 
+chat_history = st.session_state.get("chat_history", [])
 
-chat_container = st.container()
+chat_container = st.container()  
 
 for message in chat_history:
   # Display message
-  role = message["role"] 
-  content = message["content"]
+  role = message["role"]
+  content = message["content"] 
   
   with chat_container.empty():
     st.markdown(f"**{role}:** {content}")
 
-# Get user input
-user_input = st.text_input("Ask the assistant")  
+# Get user input  
+user_input = st.text_input("Ask the assistant")   
 
 if user_input:
 
-  # Process input  
+  # Process input
   chat_history.append({"role": "user", "content": user_input})
-  
+   
+  # Display input  
   with chat_container.empty():
     st.markdown(f"**user:** {user_input}")
 
   # Get response
   with st.spinner("Thinking..."):
-    response = model.ask(user_input, history=chat_history)  
+    convo = model.start_chat(history=chat_history)
+    response = convo.last.text
 
-  # Process response
-  chat_history.append({"role": "assistant", "content": response })
+  # Process response 
+  chat_history.append({"role": "assistant", "content": response})
   
+  # Display response
   with chat_container.empty():
     st.markdown(f"**assistant:** {response}")
-  
+    
   # Update session state
   st.session_state["chat_history"] = chat_history