Update app.py

app.py

@@ -4,68 +4,74 @@ import google.generativeai as genai
 # Configure API key
 genai.configure(api_key="AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM")
 
-# Create model
+# Create model object
 model = genai.GenerativeModel(
-    model_name="gemini-pro",
-    generation_config={
-        "temperature": 0.9,
-        "max_output_tokens": 2048
+  model_name="gemini-pro",
+  generation_config={
+    "temperature": 0.9,
+    "max_output_tokens": 2048
+  },
+  safety_settings=[
+    {
+      "category": "HARM_CATEGORY_HARASSMENT",
+      "threshold": "BLOCK_MEDIUM_AND_ABOVE"
     },
-    safety_settings=[
-        {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
-        {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
-        {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
-        {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"}
-    ]
+    {
+      "category": "HARM_CATEGORY_HATE_SPEECH",
+      "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+    },
+    {
+      "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+      "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+    },
+    {
+      "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+      "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+    }
+  ]
 )
 
-# Chatbot interface
+# Create chatbot interface
 st.title("Gemini API Chatbot")
 
+# Get chat history from session state
 chat_history = st.session_state.get("chat_history", [])
+
+# Create container for chat messages
 chat_container = st.container()
 
+# Loop over chat history and display messages
 for message in chat_history:
-    role = message.role 
-    text = message.parts[0].text
-    
-    chat_container.markdown(f"**{role}:** {text}")
-    
+  # Get message role and text
+  role = message.role
+  text = message.parts[0].text
+
+  # Display message with markdown
+  chat_container.markdown(f"**{role}:** {text}")
+
+# Get user input from text box
 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(user_input)
 
-    # Create user message
-    user_message = genai.Content(
-        parts=[genai.Part(text=user_input)],
-        role="user"
-    )
-
-    # Add user message to chat history
-    chat_history.append(user_message)
+  # Display user message with markdown
+  chat_container.markdown(f"**user:** {user_input}")
 
-    # Display user message
-    chat_container.markdown(f"**user:** {user_input}")
+  # Get model response with start_chat method
+  with st.spinner("Thinking..."):
+    response = model.start_chat(user_input)
 
-    # Get model response
-    with st.spinner("Thinking..."):
-        convo = model.start_chat(chat_history)
-        response = convo.last
+  # Get response text from response object
+  response_text = response.parts[0].text
 
-    # Get response text
-    response_text = response.parts[0].text
-    
-    # Create response message 
-    response_message = genai.Content(
-        parts=[genai.Part(text=response_text)],
-        role="assistant"
-    )
+  # Add response message to chat history
+  chat_history.append(response_text)
 
-    # Add response to chat history
-    chat_history.append(response_message)
+  # Display response message with markdown
+  chat_container.markdown(f"**assistant:** {response_text}")
 
-    # Display response
-    chat_container.markdown(f"**assistant:** {response_text}")
-    
-    # Update session state
-    st.session_state["chat_history"] = chat_history
+  # Update session state with chat history
+  st.session_state["chat_history"] = chat_history