Update app.py

app.py

@@ -21,6 +21,37 @@ api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
 # Configure the API key
 genai.configure(api_key=api_key)
 
+# Set up the model
+generation_config = {
+  "temperature": 0.9,
+  "top_p": 1,
+  "top_k": 1,
+  "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"
+  }
+]
+
+model = genai.GenerativeModel(model_name="gemini-pro",
+                              generation_config=generation_config,
+                              safety_settings=safety_settings)
+
 # Create chatbot interface
 st.title("Gemini API Chatbot")
 
@@ -38,10 +69,10 @@ 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({"role": "user", "content": user_input})
+  chat_history.append({"role": "user", "parts": [user_input]})
 
-  # Create model object
-  model = genai.GenerativeModel(model_name="gemini-pro")
+  # Display user message with markdown
+  st.markdown(f"**You:** {user_input}")
 
   # Get model response with generate_content method
   with st.spinner("Thinking..."):
@@ -51,7 +82,10 @@ if user_input:
   response_text = response.text
 
   # Add response message to chat history
-  chat_history.append({"role": "assistant", "content": response_text})
+  chat_history.append({"role": "assistant", "parts": [response_text]})
+
+  # Display response message with markdown
+  st.markdown(f"**Gemini Bot:** {response_text}")
 
   # Update session state with chat history
   st.session_state["chat_history"] = chat_history
@@ -74,7 +108,7 @@ if st.button("Clear History"):
 
 # Save chat history to database
 for message in chat_history:
-    c.execute("INSERT INTO history VALUES (?, ?)", (message["role"], message["content"]))
+    c.execute("INSERT INTO history VALUES (?, ?)", (message["role"], message["parts"][0]))
 conn.commit()
 
 # Close the connection