Update app.py

app.py

@@ -162,25 +162,24 @@ try:
        )
 except Exception as e:
    st.write(f"An error occurred: {e}")
-   # No need to return here
-
-    # Add model response to chat history
-    st.session_state["chat_history"].append({"role": "model", "parts": [{"text": response}]})
-
-    # Display chat history
-    for message in st.session_state["chat_history"]:
-        r, t = message["role"], message["parts"][0]["text"]
-        st.markdown(f"**{r.title()}:** {t}")
-
-    # Save chat history to database
-    for message in st.session_state["chat_history"]:
-        if len(st.session_state["chat_history"]) % 2 == 0:
-            role = "user"
-        else:
-            role = "model" 
-        text = str(message["parts"][0]["text"]) # Ensure the text is a string
-        c.execute("INSERT INTO history VALUES (?, ?)", (role, text)) 
-    conn.commit() 
-
-    # Clear user input
-    st.session_state.user_input = ""
+
+# Add model response to chat history
+st.session_state["chat_history"].append({"role": "model", "parts": [{"text": response}]})
+
+# Display chat history
+for message in st.session_state["chat_history"]:
+   r, t = message["role"], message["parts"][0]["text"]
+   st.markdown(f"**{r.title()}:** {t}")
+
+# Save chat history to database
+for message in st.session_state["chat_history"]:
+   if len(st.session_state["chat_history"]) % 2 == 0:
+       role = "user"
+   else:
+       role = "model" 
+   text = str(message["parts"][0]["text"]) # Ensure the text is a string
+   c.execute("INSERT INTO history VALUES (?, ?)", (role, text)) 
+conn.commit() 
+
+# Clear user input
+st.session_state.user_input = ""