Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-import google.generativeai as genai 
+import google.generativeai as genai
 import sqlite3
 
 conn = sqlite3.connect('chat_history.db')
@@ -10,62 +10,69 @@ c.execute('''
           (role TEXT, message TEXT)  
           ''')
           
-api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
+api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM" 
 genai.configure(api_key=api_key)
 
 generation_config = {
-  "temperature": 0.9,    
-  "max_output_tokens": 2000,
+  "temperature": 0.9,
+  "max_output_tokens": 2000  
 }
 
-safety_settings = [] 
+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
 )
-   
-st.title("Chatbot")
-   
-chat_history = st.session_state.get("chat_history", [])  
 
-if len(chat_history) % 2 == 0:
-  role = "user"
-else:
-  role = "model"
-  
+st.title("Chatbot")  
+
+chat_history = st.session_state.get("chat_history", [])
+
 displayed_msgs = []
 for message in chat_history:
-  if message not in displayed_msgs:
-    r, t = message["role"], message["parts"][0]["text"]
-    st.markdown(f"**{r.title()}:** {t}")
-    displayed_msgs.append(message)
+  r, t = message["role"], message["parts"][0]["text"]
+  st.markdown(f"**{r.title()}:** {t}")
+  displayed_msgs.append(message)
+  
+user_input = st.text_input("")  
 
 if user_input: 
-  # Existing code
+
+  if len(chat_history) % 2 == 0:
+    role = "user"
+  else:
+    role = "model"
+    
+  chat_history.append({"role": role, "parts": [{"text": user_input}]})
+  
+  if role == "user":
+    response = model.generate_content(chat_history)
+    response_text = response.text  
+    chat_history.append({"role": "model", "parts": [{"text": response_text}]})
 
   st.session_state["chat_history"] = chat_history
   
   displayed_msgs = []
   for message in chat_history:
-    if message not in displayed_msgs:
+    if message not in displayed_msgs:  
       r, t = message["role"], message["parts"][0]["text"]
       st.markdown(f"**{r.title()}:** {t}")
-      displayed_msgs.append(message)
-
-if st.button("Display History"):
-  displayed_msgs = [] 
-  c.execute("SELECT * FROM history")
-  rows = c.fetchall()
-
-  for row in rows:
-    if {"role": row[0], "parts": [{"text": row[1]}]} not in displayed_msgs:
-      st.markdown(f"**{row[0].title()}:** {row[1]}") 
-      displayed_msgs.append({"role": row[0], "parts": [{"text": row[1]}]})
+      displayed_msgs.append(message)  
         
+  if st.button("Display History"):    
+    displayed_msgs = [] 
+    c.execute("SELECT * FROM history")
+    rows = c.fetchall()
+
+    for row in rows:
+      if {"role": row[0], "parts": [{"text": row[1]}]} not in displayed_msgs:  
+        st.markdown(f"**{row[0].title()}:** {row[1]}")
+        displayed_msgs.append({"role": row[0], "parts": [{"text": row[1]]})
+              
 for message in chat_history:
-  if {"role": message["role"], "parts": [message["parts"][0]]} not in rows:  
+  if {"role": message["role"], "parts": [message["parts"][0]]} not in rows:   
     c.execute("INSERT INTO history VALUES (?, ?)",
               (message["role"], message["parts"][0]["text"])) 
     conn.commit()