new

app.py

@@ -388,32 +388,39 @@ with gr.Blocks() as demo:
         combined_txt_results = gr.Textbox(label="Combined Repo Files", lines=20)
         llm_output_txt_results = gr.Textbox(label="LLM Analysis Output", lines=10)
         back_to_start_btn4 = gr.Button("Back to Start")
-        # Batch analyze and show top 3 only on results page
+        go_to_batch_btn = gr.Button("Go to Batch Analysis Page", visible=True)
+
+    # --- Page 5: Batch Analysis Page ---
+    with gr.Column(visible=False) as batch_page:
+        gr.Markdown("## Batch Analysis & Top 3 Selection")
         batch_btn = gr.Button("Batch Analyze All & Select Top 3", visible=True)
         batch_info_txt = gr.Textbox(label="All Repo Analyses", lines=10)
         top3_txt = gr.Textbox(label="Top 3 Repo IDs", lines=1)
-        show_top3_page_btn = gr.Button("Show Top 3 Repos on New Page", visible=True)
+        show_top3_chat_btn = gr.Button("Show Top 3 Repo IDs in Chat", visible=True)
+        back_to_results_from_batch_btn = gr.Button("Back to Results")
 
-    # --- Page 5: Top 3 Repos Page ---
+    # --- Page 6: Top 3 Repos Page ---
     with gr.Column(visible=False) as top3_page:
         gr.Markdown("## Top 3 Recommended Repos")
         top3_df = gr.Dataframe(headers=["repo id"], datatype=["str"])
         back_to_results_btn = gr.Button("Back to Results")
 
     # Navigation logic
-    option_a_btn.click(go_to_input, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, top3_page])
+    option_a_btn.click(go_to_input, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, batch_page, top3_page])
     option_b_btn.click(
-        lambda: (gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), [["", CHATBOT_INITIAL_MESSAGE]]),
+        lambda: (gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), [["", CHATBOT_INITIAL_MESSAGE]]),
         inputs=None,
-        outputs=[start_page, input_page, chatbot_page, results_page, top3_page, state]
+        outputs=[start_page, input_page, chatbot_page, results_page, batch_page, top3_page, state]
     )
-    next_btn.click(go_to_analysis, inputs=None, outputs=[input_page, analysis_page, chatbot_page, results_page, top3_page])
-    back_btn.click(go_to_input, inputs=None, outputs=[input_page, analysis_page, chatbot_page, results_page, top3_page])
-    back_to_start_btn.click(go_to_start, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, top3_page])
-    back_to_start_btn2.click(go_to_start, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, top3_page])
-    back_to_start_btn3.click(go_to_start, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, top3_page])
-    back_to_start_btn4.click(go_to_start, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, top3_page])
-    back_to_results_btn.click(lambda: (gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)), inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, top3_page])
+    next_btn.click(go_to_analysis, inputs=None, outputs=[input_page, analysis_page, chatbot_page, results_page, batch_page, top3_page])
+    back_btn.click(go_to_input, inputs=None, outputs=[input_page, analysis_page, chatbot_page, results_page, batch_page, top3_page])
+    back_to_start_btn.click(go_to_start, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, batch_page, top3_page])
+    back_to_start_btn2.click(go_to_start, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, batch_page, top3_page])
+    back_to_start_btn3.click(go_to_start, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, batch_page, top3_page])
+    back_to_start_btn4.click(go_to_start, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, batch_page, top3_page])
+    go_to_batch_btn.click(lambda: (gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)), inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, batch_page, top3_page])
+    back_to_results_from_batch_btn.click(lambda: (gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)), inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, batch_page, top3_page])
+    back_to_results_btn.click(lambda: (gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)), inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, batch_page, top3_page])
 
     # Keyword and repo input logic
     keyword_btn.click(keyword_search_and_update, inputs=keyword_input, outputs=df_box)
@@ -438,19 +445,20 @@ with gr.Blocks() as demo:
     def go_to_results_from_chatbot(keywords):
         # Use the keywords to search and update the CSV, then display the DataFrame
         df = use_keywords_to_search_and_update_csv(keywords)
-        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), df
+        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), df
 
     send_btn.click(user_send, inputs=[user_input, state], outputs=[chatbot, state, user_input])
     end_btn.click(end_chat, inputs=state, outputs=keywords_output)
     go_to_results_btn.click(
         go_to_results_from_chatbot,
         inputs=keywords_output,
-        outputs=[chatbot_page, input_page, analysis_page, results_page, top3_page, results_df]
+        outputs=[chatbot_page, input_page, analysis_page, results_page, batch_page, top3_page, results_df]
     )
 
     # Add logic for the new button on results_page
     analyze_next_btn.click(show_combined_repo_and_llm, inputs=None, outputs=[combined_txt_results, llm_output_txt_results, results_df])
     batch_btn.click(batch_analyze_and_select_top, inputs=None, outputs=[batch_info_txt, top3_txt, df_output])
+    show_top3_chat_btn.click(batch_analyze_and_select_top_for_chat, inputs=[state], outputs=[state])
 
     def show_top3_page():
         # Run batch analysis, get top 3, save to CSV, and return DataFrame
@@ -465,8 +473,8 @@ with gr.Blocks() as demo:
             top3_ids = []
         top3_df_data = pd.DataFrame({"repo id": top3_ids})
         top3_df_data.to_csv("top3_repos.csv", index=False)
-        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), top3_df_data
+        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), top3_df_data
 
-    show_top3_page_btn.click(show_top3_page, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, top3_page, top3_df])
+    show_top3_page_btn.click(show_top3_page, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, batch_page, top3_page, top3_df])
 
 demo.launch()

chatbot_page.py

@@ -48,6 +48,7 @@ def extract_keywords_from_conversation(history):
         "You are an expert at helping users find open-source repos on Hugging Face. "
         "Given a conversation, extract about 5 keywords that would be most useful for searching Hugging Face repos to find the most relevant results for the user. "
         "Return only the keywords as a comma-separated list."
+        "Use keywords that are specific to the user's use case and features they are looking for."
     )
     user_prompt = (
         "Conversation:\n" + conversation + "\n\nExtract about 5 keywords for Hugging Face repo search."