Update app.py

app.py

@@ -284,19 +284,19 @@ def streamlit_app():
                 if db:
                     st.write("Models have been processed and added to the database.")
                     
-                    user_query = st.text_input("Ask a question about the biomodels:")
-                    
-                    if user_query:
-                        if 'previous_context' not in st.session_state:
-                            st.session_state.previous_context = ""
-                        
-                        # Stream the response incrementally for the second generation
-                        response = generate_response(db, user_query, st.session_state.previous_context)
-                        st.write(f"Final Response: {response}")
-                        
-                        st.session_state.previous_context += f"{response}\n"
-        else:
-            st.write("No models found for the given search query.")
+    # Move user query input outside of the model search block
+    user_query = st.text_input("Ask a question about the biomodels:")
+
+    # Ensure that query submission and response generation occur properly
+    if user_query and db:
+        if 'previous_context' not in st.session_state:
+            st.session_state.previous_context = ""
+        
+        # Stream the response incrementally for the second generation
+        response = generate_response(db, user_query, st.session_state.previous_context)
+        st.write(f"Final Response: {response}")
+        
+        st.session_state.previous_context += f"{response}\n"
 
 if __name__ == "__main__":
     streamlit_app()