message_init

app.py

@@ -389,17 +389,24 @@ with gr.Blocks() as demo:
         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_chat_btn = gr.Button("Show Top 3 Repo IDs in Chat", visible=True)
+        show_top3_page_btn = gr.Button("Show Top 3 Repos on New Page", visible=True)
+
+    # --- Page 5: 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])
-    option_b_btn.click(go_to_chatbot, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page])
-    next_btn.click(go_to_analysis, inputs=None, outputs=[input_page, analysis_page, chatbot_page, results_page])
-    back_btn.click(go_to_input, inputs=None, outputs=[input_page, analysis_page, chatbot_page, results_page])
-    back_to_start_btn.click(go_to_start, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page])
-    back_to_start_btn2.click(go_to_start, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page])
-    back_to_start_btn3.click(go_to_start, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page])
-    back_to_start_btn4.click(go_to_start, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page])
+    option_a_btn.click(go_to_input, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, top3_page])
+    option_b_btn.click(go_to_chatbot, 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, 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])
 
     # Keyword and repo input logic
     keyword_btn.click(keyword_search_and_update, inputs=keyword_input, outputs=df_box)
@@ -424,20 +431,35 @@ 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), df
+        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), 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, results_df]
+        outputs=[chatbot_page, input_page, analysis_page, results_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 analyze and show top 3 logic (now only on results page)
     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
+        all_info_str, top3_str, df = batch_analyze_and_select_top()
+        import pandas as pd
+        import ast
+        try:
+            top3_ids = ast.literal_eval(top3_str)
+            if isinstance(top3_ids, str):
+                top3_ids = [top3_ids]
+        except Exception:
+            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
+
+    show_top3_page_btn.click(show_top3_page, inputs=None, outputs=[start_page, input_page, chatbot_page, results_page, top3_page, top3_df])
 
 demo.launch()

chatbot_page.py

@@ -69,10 +69,9 @@ def extract_keywords_from_conversation(history):
 with gr.Blocks() as chatbot_demo:
     gr.Markdown("## Repo Recommendation Chatbot")
     chatbot = gr.Chatbot()
-    # Initial assistant message and dummy question
+    # Initial assistant message only
     initial_message = "Hello! Please tell me about your ideal Hugging Face repo. What use case, preferred language, or features are you looking for?"
-    dummy_question = "What kind of project or task are you interested in?"
-    state = gr.State([["", initial_message], ["", dummy_question]])  # Start with assistant message and question
+    state = gr.State([["", initial_message]])  # Only the initial assistant message
     user_input = gr.Textbox(label="Your message", placeholder="Describe your ideal repo or answer the assistant's questions...")
     send_btn = gr.Button("Send")
     end_btn = gr.Button("End Chat and Extract Keywords")
@@ -87,8 +86,13 @@ with gr.Blocks() as chatbot_demo:
         keywords = extract_keywords_from_conversation(history)
         return keywords
 
+    # Reset state to initial message when chatbot page is loaded
+    def reset_chat_state():
+        return [["", initial_message]]
+
     send_btn.click(user_send, inputs=[user_input, state], outputs=[chatbot, state, user_input])
     end_btn.click(end_chat, inputs=state, outputs=keywords_output)
+    chatbot_demo.load(reset_chat_state, inputs=None, outputs=state)
 
 if __name__ == "__main__":
     chatbot_demo.launch()