Update app.py

app.py

@@ -1,92 +1,71 @@
-# Import modules
 import streamlit as st
 import google.generativeai as genai
 
 # Configure API key
 genai.configure(api_key="AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM")
 
-# Create model object
+# Create model
 model = genai.GenerativeModel(
-  model_name="gemini-pro", # Specify model name
-  generation_config={
-    "temperature": 0.9, # Set generation temperature
-    "max_output_tokens": 2048 # Set maximum output tokens
-  },
-  safety_settings=[ # Set safety settings
-    {
-      "category": "HARM_CATEGORY_HARASSMENT",
-      "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+    model_name="gemini-pro",
+    generation_config={
+        "temperature": 0.9,
+        "max_output_tokens": 2048
     },
-    {
-      "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"
-    }
-  ]
+    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"}
+    ]
 )
 
-# Create chatbot interface
-st.title("Gemini API Chatbot") # Set title
+# 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:
-  # 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
+    role = message.role 
+    text = message.parts[0].text
+    
+    chat_container.markdown(f"**{role}:** {text}")
+    
 user_input = st.text_input("You")
 
-# Check if user input is not empty
 if user_input:
 
-  # Create user message object
-  user_message = genai.GenerativeContent(
-    parts=[genai.Part(text=user_input)], # Wrap text in a list
-    role="user" # Set role to user
-  )
-
-  # Add user message to chat history
-  chat_history.append(user_message)
-
-  # Display user message with markdown
-  chat_container.markdown(f"**user:** {user_input}")
-
-  # Get model response with start_chat method
-  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
-
-  # Create response message object
-  response_message = genai.GenerativeContent(
-    parts=[genai.Part(text=response_text)], # Wrap text in a list
-    role="assistant" # Set role to assistant
-  )
-
-  # Add response message to chat history
-  chat_history.append(response_message)
-
-  # Display response message with markdown
-  chat_container.markdown(f"**assistant:** {response_text}")
-
-  # Update session state with chat history
-  st.session_state["chat_history"] = chat_history
+    # 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
+    chat_container.markdown(f"**user:** {user_input}")
+
+    # Get model response
+    with st.spinner("Thinking..."):
+        convo = model.start_chat(chat_history)
+        response = convo.last
+
+    # 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 to chat history
+    chat_history.append(response_message)
+
+    # Display response
+    chat_container.markdown(f"**assistant:** {response_text}")
+    
+    # Update session state
+    st.session_state["chat_history"] = chat_history