Update app.py

Trying to update a bug in the reset button

app.py

@@ -1,129 +1,113 @@
 import streamlit as st
-import google.generativeai as genai
+import google.generativeai as genai 
 import sqlite3
 from streamlit import file_uploader
 
 # Database setup
-conn = sqlite3.connect('chat_history.db') 
-c = conn.cursor() 
+conn = sqlite3.connect('chat_history.db')
+c = conn.cursor()  
 
 c.execute('''
-          CREATE TABLE IF NOT EXISTS history  
-          (role TEXT, message TEXT)  
+          CREATE TABLE IF NOT EXISTS history
+          (role TEXT, message TEXT)
           ''')
 
 # Generative AI setup
-api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM" 
+api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"  
 genai.configure(api_key=api_key)
 
 generation_config = {
   "temperature": 0.9,
-  "max_output_tokens": 500 
+  "max_output_tokens": 500
 }
 
-safety_settings = [] 
+safety_settings = []
 
 # Streamlit UI
-st.title("Chatbot") 
+st.title("Chatbot")
 
 chat_history = st.session_state.get("chat_history", [])
 
-if len(chat_history) % 2 == 0: 
-  role = "user" 
-else:  
-  role = "model"
+if len(chat_history) % 2 == 0:
+    role = "user"
+else:
+    role = "model"
 
-for message in chat_history: 
-  r, t = message["role"], message["parts"][0]["text"] 
-  st.markdown(f"**{r.title()}:** {t}") 
-
-user_input = st.text_input("") 
-
-if user_input: 
-  chat_history.append({"role": role, "parts": [{"text": user_input}]})
-
-  if role == "user": 
-    # If only text is entered, follow the previous code
-    model_name = "gemini-pro"
-
-    model = genai.GenerativeModel(
-      model_name=model_name,
-      generation_config=generation_config,
-      safety_settings=safety_settings
-    )
-
-    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 
-
-  for message in chat_history: 
+for message in chat_history:
     r, t = message["role"], message["parts"][0]["text"] 
-    st.markdown(f"**{r.title()}:** {t}") 
-
-if st.button("Display History"): 
-    c.execute("SELECT * FROM history") 
-    rows = c.fetchall() 
-
-    for row in rows: 
-        st.markdown(f"**{row[0].title()}:** {row[1]}") 
+    st.markdown(f"**{r.title()}:** {t}")
+    
+user_input = st.text_input("")  
+if user_input:
+    chat_history.append({"role": role, "parts": [{"text": user_input}]})
+    if role == "user":
+        
+        # If only text is entered, follow the previous code  
+        model_name = "gemini-pro"
+        model = genai.GenerativeModel(
+            model_name=model_name,  
+            generation_config=generation_config,
+            safety_settings=safety_settings
+        )
         
-# Save chat history to database
-for message in chat_history: 
-    c.execute("INSERT INTO history VALUES (?, ?)", 
-            (message["role"], message["parts"][0]["text"])) 
-    conn.commit() 
+        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
+        
+for message in chat_history:
+    r, t = message["role"], message["parts"][0]["text"] 
+    st.markdown(f"**{r.title()}:** {t}")
 
-# Create a placeholder for the history display
+    
+# Create a placeholder for the history display    
 history_placeholder = st.empty()
 
 # Use the key argument for the buttons
-if st.button("Display History", key="display"): 
-    c.execute("SELECT * FROM history") 
-    rows = c.fetchall() 
-
-    # Fill the placeholder with the history
-    for row in rows: 
-        history_placeholder.markdown(f"**{row[0].title()}:** {row[1]}") 
+if st.button("Display History", key="display"):
+    c.execute("SELECT * FROM history")
+    rows = c.fetchall()
 
+    # Fill the placeholder with the history  
+    for row in rows:
+        history_placeholder.markdown(f"**{row[0].title()}:** {row[1]}")
+        
 # Define a function to clear the placeholder
 def clear_history():
     history_placeholder.empty()
-
-# Use the key argument for the buttons
+    
+# Use the key argument for the buttons    
 if st.button("Reset History", key="reset", on_click=clear_history):
     pass
 
-
 conn.close()
 
 # Separate section for image uploading
-st.title("Image Description Generator")
-
-uploaded_file = st.file_uploader("Upload an image here", type=["png", "jpg", "jpeg"])
+st.title("Image Description Generator") 
+uploaded_file = st.file_uploader("Upload an image here", type=["png", "jpg", "jpeg"])  
 
-# Text input for asking questions about the image
+# Text input for asking questions about the image:
 image_question = st.text_input("Ask something about the image:")
 
 if uploaded_file and image_question:
-   image_parts = [
-     {
-       "mime_type": uploaded_file.type,
-       "data": uploaded_file.read() 
-     },
-   ]
-
-   prompt_parts = [
-     image_question,
-     image_parts[0],
-   ]
-
-   model = genai.GenerativeModel(
-     model_name="gemini-pro-vision", 
-     generation_config=generation_config, 
-     safety_settings=safety_settings 
-   )
-   
-   response = model.generate_content(prompt_parts)
-   st.markdown(f"**Model's answer:** {response.text}")
+    image_parts = [
+        {
+            "mime_type": uploaded_file.type,
+            "data": uploaded_file.read()
+        },
+    ]
+
+    prompt_parts = [
+        image_question, 
+        image_parts[0],
+    ]
+    
+    model = genai.GenerativeModel(
+        model_name="gemini-pro-vision",  
+        generation_config=generation_config,
+        safety_settings=safety_settings
+    )
+    
+    response = model.generate_content(prompt_parts)  
+    st.markdown(f"**Model's answer:** {response.text}")