Update app.py

app.py

@@ -178,8 +178,7 @@ def create_vector_db(final_items):
                 temperature=0.1, 
                 top_p=0.9, 
                 top_k=20, 
-                stream=False,
-                verbose = True
+                stream=False
             )
     
             # Extract the generated summary text
@@ -242,8 +241,7 @@ def generate_response(db, query_text, previous_context):
         stream=True,  # Enable streaming
         temperature=0.1,
         top_p=0.9,
-        top_k=20,
-        verbose = True
+        top_k=20
     )
     
     # Use Streamlit to stream the response in real-time
@@ -303,20 +301,19 @@ def streamlit_app():
                 if db:
                     st.write("Models have been processed and added to the database.")
                     
-    # Move user query input outside of the model search block
-    user_query = st.text_input("Ask a question about the biomodels:")
+    # Check if the database is created before showing the query input
+    if db:
+        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 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"
 
 if __name__ == "__main__":
-    streamlit_app()
-
+    streamlit_app()