Update app.py

app.py

@@ -263,7 +263,8 @@ def streamlit_app():
     st.title("BioModelsRAG")
     
     search_str = st.text_input("Enter search query:")
-    
+
+    # Keep the search input field visible even after submission
     if search_str:
         models = search_models(search_str)
         
@@ -303,17 +304,20 @@ def streamlit_app():
                     
     # Check if the database is created before showing the query input
     if db:
-        user_query = st.text_input("Ask a question about the biomodels:")
+        user_query = st.text_input("Ask a question about the biomodels:", key="user_query")
 
         if user_query:
             if 'previous_context' not in st.session_state:
                 st.session_state.previous_context = ""
             
-            # Stream the response incrementally for the second generation
+            # Placeholder for indicating that response generation has started
+            response_placeholder = st.empty()
+            response_placeholder.write("Response generation is beginning...")
+
+            # Generate and display response
             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()
+            # Clear the "Response generation is beginning..." placeholder after generation
+            response_placeholder.empty()