Updated the demo Function

Components/para_agent.py

@@ -121,7 +121,9 @@ def demo_file_load():
 
     retriever = db.as_retriever(similarity_score_threshold=0.8)
     print(retriever,"retrived document indemo")
-
+    query1 = "What is LLM?"
+    results = retriever.get_relevant_documents(query1)
+    print(results)
     prompt_template = """
     CONTEXT: {context}
     QUESTION: {question}"""

app.py

@@ -139,8 +139,8 @@ with st.form(key="my_form"):
 if submit_button  and generate_questions_checkbox and (uploaded_file or use_demo):
     query = f"Generate 5 flashcard questions based Context: {query}"
     st.write(query)
-    if agent:
-        response, Source = agent.ask(query)
+    if st.session_state.agent:
+        response, Source = st.session_state.agent.ask(query)
         add_query_response(query, response)
 
         # Uncheck the checkbox after processing
@@ -165,7 +165,7 @@ if submit_button  and generate_questions_checkbox and (uploaded_file or use_demo
         st.write("No documents found.")
 
 # Modify the query processing section
-if submit_button and query and not uploaded_video_file:  # Check if button is pressed
+if submit_button and query and not uploaded_video_file and not generate_questions_checkbox:  # Check if button is pressed
     if st.session_state.agent:
         response, Source = st.session_state.agent.ask(query)
         add_query_response(query, response)
@@ -218,25 +218,3 @@ else:
 # Add this after session state initialization
 if 'needs_cleanup' not in st.session_state:
     st.session_state.needs_cleanup = False
-
-def cleanup_temp_files():
-    """Clean up temporary files when the session ends"""
-    try:
-        # Clean up the temporary video files
-        temp_dir = tempfile.gettempdir()
-        for filename in os.listdir(temp_dir):
-            if filename.endswith(('.mp4', '.avi', '.mov', '.wav')):
-                filepath = os.path.join(temp_dir, filename)
-                os.remove(filepath)
-                
-        # Clean up any temporary audio files in current directory
-        if os.path.exists("temp_audio.wav"):
-            os.remove("temp_audio.wav")
-            
-        # Clean up dataset directory but keep user storage
-        if os.path.exists("dataset"):
-            shutil.rmtree("dataset")
-            
-    except Exception as e:
-        st.error(f"Error during cleanup: {e}")
-