Update app.py

app.py

@@ -1,96 +1,92 @@
+# Import modules
 import streamlit as st
-import pandas as pd
-import numpy as np
 import google.generativeai as genai
 
-# API key 
-genai.configure(api_key="AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LwNM")
-
-# Model settings
-generation_config = {
-  "temperature": 0.9,
-  "max_output_tokens": 2048    
-}
-
-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"
-  }
-]
+# Configure API key
+genai.configure(api_key="YOUR_API_KEY_HERE")
 
+# Create model object
 model = genai.GenerativeModel(
-  model_name="gemini-pro", 
-  generation_config=generation_config,
-  safety_settings=safety_settings
+  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"
+    },
+    {
+      "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
-st.title("Gemini API Chatbot")
+# Create chatbot interface
+st.title("Gemini API Chatbot") # Set title
 
-chat_history = st.session_state.get("chat_history", []) 
+# 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:
-  # Display message
-  if message.parts:
-    role = "assistant" if message.role == "system" else "user"
-    text = message.parts[0].text 
-  else:
-    role = message.role
-    text = message.content
-  
+  # Get message role and text
+  role = message.role
+  text = message.parts[0].text
+
+  # Display message with markdown
   chat_container.markdown(f"**{role}:** {text}")
 
-# Input  
+# Get user input from text box
 user_input = st.text_input("You")
 
+# Check if user input is not empty
 if user_input:
 
-  # Create message
+  # Create user message object
   user_message = genai.GenerativeContent(
-      parts=[genai.Part(text=user_input)], # Wrap in a list
-      role=genai.Role.USER
+    parts=[genai.Part(text=user_input)], # Wrap text in a list
+    role="user" # Set role to user
   )
 
-  # Append message  
+  # Add user message to chat history
   chat_history.append(user_message)
 
-  # Display message
+  # Display user message with markdown
   chat_container.markdown(f"**user:** {user_input}")
 
-  # Get response
+  # Get model response with start_chat method
   with st.spinner("Thinking..."):
     convo = model.start_chat(chat_history)
     response = convo.last
 
-  # Extract response text
+  # Get response text from response object
   response_text = response.parts[0].text
-  
-  # Create response message
-  assistant_message = genai.GenerativeContent(
-      parts=[genai.Part(text=response_text)], # Wrap in a list
-      role=genai.Role.ASSISTANT
+
+  # 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
   )
 
-  # Append response 
-  chat_history.append(assistant_message)
+  # Add response message to chat history
+  chat_history.append(response_message)
 
-  # Display response 
+  # Display response message with markdown
   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