Update app.py

corrected some bugs in the select paper for updating it realtime

app.py

@@ -94,34 +94,34 @@ except Exception as e:
     raise
 
 # Hybrid search function
-def get_relevant_papers(query, top_k=5):
+def get_relevant_papers(query):
     if not query.strip():
-        return []
+        return [], "Please enter a search query."
     try:
         query_embedding = generate_embeddings_sci_bert([query])
-        distances, indices = faiss_index.search(query_embedding.astype(np.float32), top_k)
+        distances, indices = faiss_index.search(query_embedding.astype(np.float32), 5)
         tokenized_query = query.split()
         bm25_scores = bm25.get_scores(tokenized_query)
-        bm25_top_indices = np.argsort(bm25_scores)[::-1][:top_k]
+        bm25_top_indices = np.argsort(bm25_scores)[::-1][:5]
         combined_indices = list(set(indices[0]) | set(bm25_top_indices))
         ranked_results = sorted(combined_indices, key=lambda idx: -bm25_scores[idx])
         papers = []
-        for i, index in enumerate(ranked_results[:top_k]):
+        for i, index in enumerate(ranked_results[:5]):
             paper = df.iloc[index]
             papers.append(f"{i+1}. {paper['title']} - Abstract: {paper['cleaned_abstract'][:200]}...")
-        return papers
+        return papers, "Search completed."
     except Exception as e:
         logger.error(f"Search failed: {e}")
-        return ["Search failed. Please try again."]
+        return [], "Search failed. Please try again."
 
 # GPT-2 QA function
 def answer_question(paper, question, history):
     if not paper:
-        return [("Please select a paper first!", "")], history
+        return [(question, "Please select a paper first!")], history
     if not question.strip():
         return [(question, "Please ask a question!")], history
     if question.lower() in ["exit", "done"]:
-        return [("Conversation ended. Select a new paper or search again!", "")], []
+        return [("Conversation ended.", "Select a new paper or search again!")], []
 
     try:
         # Extract title and abstract
@@ -150,7 +150,7 @@ def answer_question(paper, question, history):
         response = response[len(context):].strip()
         
         history.append((question, response))
-        return history, history  # Return updated history for Chatbot
+        return history, history
     except Exception as e:
         logger.error(f"QA failed: {e}")
         history.append((question, "Sorry, I couldn’t process that. Try again!"))
@@ -173,9 +173,18 @@ with gr.Blocks(
             query_input = gr.Textbox(label="Enter your search query", placeholder="e.g., machine learning in healthcare")
             search_btn = gr.Button("Search")
             paper_dropdown = gr.Dropdown(label="Select a Paper", choices=[], interactive=True)
+            search_status = gr.Textbox(label="Search Status", interactive=False)
+            
+            # State to store paper choices
+            paper_choices_state = gr.State([])
+
             search_btn.click(
                 fn=get_relevant_papers,
                 inputs=query_input,
+                outputs=[paper_choices_state, search_status]
+            ).then(
+                fn=lambda choices: gr.update(choices=choices, value=None),
+                inputs=paper_choices_state,
                 outputs=paper_dropdown
             )
         
@@ -190,11 +199,15 @@ with gr.Blocks(
             # State to store conversation history
             history_state = gr.State([])
 
-            # Update selected paper
+            # Update selected paper and reset history
             paper_dropdown.change(
-                fn=lambda x: (x, []),  # Reset history when new paper selected
+                fn=lambda x: (x, []),
                 inputs=paper_dropdown,
                 outputs=[selected_paper, history_state]
+            ).then(
+                fn=lambda: [],
+                inputs=None,
+                outputs=chatbot
             )
             
             # Handle chat
@@ -205,7 +218,7 @@ with gr.Blocks(
             ).then(
                 fn=lambda: "",
                 inputs=None,
-                outputs=question_input  # Clear question input after sending
+                outputs=question_input  # Clear input
             )
 
 # Launch the app