Update app.py

app.py

@@ -245,9 +245,8 @@ def create_ui() -> gr.Blocks:
                         lines=5,
                         placeholder="repo1, repo2\nrepo3"
                     )
-                    with gr.Row():
-                        submit_btn = gr.Button("Submit Repo IDs", variant="primary")
-                        submit_loading = gr.Loading(visible=False)
+                    submit_btn = gr.Button("Submit Repo IDs", variant="primary")
+                    submit_status = gr.Textbox(label="Status", visible=False)
                     
                 with gr.Column():
                     gr.Markdown("### Or Search by Keywords")
@@ -256,9 +255,8 @@ def create_ui() -> gr.Blocks:
                         lines=3,
                         placeholder="Enter keywords separated by commas"
                     )
-                    with gr.Row():
-                        search_btn = gr.Button("Search by Keywords", variant="primary")
-                        search_loading = gr.Loading(visible=False)
+                    search_btn = gr.Button("Search by Keywords", variant="primary")
+                    search_status = gr.Textbox(label="Status", visible=False)
                     
             df_output = gr.Dataframe(
                 headers=["repo id", "strength", "weaknesses", "speciality", "relevance rating"],
@@ -266,7 +264,7 @@ def create_ui() -> gr.Blocks:
             )
             with gr.Row():
                 analyze_btn = gr.Button("Start Analysis", variant="primary")
-                analyze_loading = gr.Loading(visible=False)
+                analyze_status = gr.Textbox(label="Status", visible=False)
                 compare_btn = gr.Button("Compare Repositories", variant="secondary")
             
         # Analysis Page
@@ -286,10 +284,10 @@ def create_ui() -> gr.Blocks:
                     summary_output = gr.Textbox(label="Analysis Summary", lines=10)
             with gr.Row():
                 next_btn = gr.Button("Analyze Next Repository", variant="primary")
-                next_loading = gr.Loading(visible=False)
+                next_status = gr.Textbox(label="Status", visible=False)
                 finish_btn = gr.Button("Finish Analysis", variant="secondary")
                 export_btn = gr.Button("Export Results", variant="secondary")
-                export_loading = gr.Loading(visible=False)
+                export_status = gr.Textbox(label="Status", visible=False)
             
         # Comparison Page
         with gr.Group(visible=False) as comparison_page:
@@ -337,9 +335,9 @@ def create_ui() -> gr.Blocks:
             )
             with gr.Row():
                 send_btn = gr.Button("Send", variant="primary")
-                send_loading = gr.Loading(visible=False)
+                send_status = gr.Textbox(label="Status", visible=False)
                 end_chat_btn = gr.Button("End Chat", variant="secondary")
-                end_chat_loading = gr.Loading(visible=False)
+                end_chat_status = gr.Textbox(label="Status", visible=False)
                 
         # Results Page
         with gr.Group(visible=False) as results_page:
@@ -394,99 +392,99 @@ def create_ui() -> gr.Blocks:
             outputs=[start_page, input_page, analysis_page, chatbot_page, results_page, help_page, comparison_page, history_page]
         )
         
-        # Modified event handlers with loading states
-        def process_repo_input_with_loading(text: str, state: AppState) -> Tuple[pd.DataFrame, gr.update]:
-            """Process repo input with loading state."""
-            return process_repo_input(text, state), gr.update(visible=False)
+        # Modified event handlers with status updates
+        def process_repo_input_with_status(text: str, state: AppState) -> Tuple[pd.DataFrame, gr.update]:
+            """Process repo input with status update."""
+            return process_repo_input(text, state), gr.update(value="", visible=False)
             
-        def keyword_search_with_loading(keyword: str, state: AppState) -> Tuple[pd.DataFrame, gr.update]:
-            """Search keywords with loading state."""
-            return keyword_search_and_update(keyword, state), gr.update(visible=False)
+        def keyword_search_with_status(keyword: str, state: AppState) -> Tuple[pd.DataFrame, gr.update]:
+            """Search keywords with status update."""
+            return keyword_search_and_update(keyword, state), gr.update(value="", visible=False)
             
-        def analyze_with_loading(state: AppState) -> Tuple[str, str, pd.DataFrame, gr.update]:
-            """Analyze with loading state."""
-            return *show_combined_repo_and_llm(state), gr.update(visible=False)
+        def analyze_with_status(state: AppState) -> Tuple[str, str, pd.DataFrame, gr.update]:
+            """Analyze with status update."""
+            return *show_combined_repo_and_llm(state), gr.update(value="", visible=False)
             
-        def send_message_with_loading(user_message: str, history: List[Dict[str, str]], state: AppState) -> Tuple[List[Dict[str, str]], str, gr.update]:
-            """Send message with loading state."""
+        def send_message_with_status(user_message: str, history: List[Dict[str, str]], state: AppState) -> Tuple[List[Dict[str, str]], str, gr.update]:
+            """Send message with status update."""
             if not user_message:
-                return history, "", gr.update(visible=False)
+                return history, "", gr.update(value="", visible=False)
             history.append({"role": "user", "content": user_message})
             response = chat_with_user(user_message, history, CHATBOT_SYSTEM_PROMPT)
             history.append({"role": "assistant", "content": response})
-            return history, "", gr.update(visible=False)
+            return history, "", gr.update(value="", visible=False)
             
-        def end_chat_with_loading(history: List[Dict[str, str]], state: AppState) -> Tuple[List[str], gr.update, gr.update]:
+        def end_chat_with_status(history: List[Dict[str, str]], state: AppState) -> Tuple[List[str], gr.update, gr.update]:
             """End chat and extract keywords."""
             if not history:
-                return [], gr.update(visible=True), gr.update(visible=False)
+                return [], gr.update(visible=True), gr.update(value="", visible=False)
             keywords = extract_keywords_from_conversation(history)
             state.generated_keywords = keywords
-            return keywords, gr.update(visible=True), gr.update(visible=False)
+            return keywords, gr.update(visible=True), gr.update(value="", visible=False)
             
-        def export_with_loading(df: pd.DataFrame) -> Tuple[str, gr.update]:
-            """Export with loading state."""
-            return export_results(df), gr.update(visible=False)
+        def export_with_status(df: pd.DataFrame) -> Tuple[str, gr.update]:
+            """Export with status update."""
+            return export_results(df), gr.update(value="", visible=False)
             
-        # Update event handlers with loading states
+        # Update event handlers with status updates
         submit_btn.click(
-            fn=lambda: gr.update(visible=True),
+            fn=lambda: gr.update(value="Processing...", visible=True),
             inputs=[],
-            outputs=[submit_loading]
+            outputs=[submit_status]
         ).then(
-            fn=process_repo_input_with_loading,
+            fn=process_repo_input_with_status,
             inputs=[repo_id_input, state],
-            outputs=[df_output, submit_loading]
+            outputs=[df_output, submit_status]
         )
         
         search_btn.click(
-            fn=lambda: gr.update(visible=True),
+            fn=lambda: gr.update(value="Searching...", visible=True),
             inputs=[],
-            outputs=[search_loading]
+            outputs=[search_status]
         ).then(
-            fn=keyword_search_with_loading,
+            fn=keyword_search_with_status,
             inputs=[keyword_input, state],
-            outputs=[df_output, search_loading]
+            outputs=[df_output, search_status]
         )
         
         next_btn.click(
-            fn=lambda: gr.update(visible=True),
+            fn=lambda: gr.update(value="Analyzing...", visible=True),
             inputs=[],
-            outputs=[next_loading]
+            outputs=[next_status]
         ).then(
-            fn=analyze_with_loading,
+            fn=analyze_with_status,
             inputs=[state],
-            outputs=[content_output, summary_output, df_output, next_loading]
+            outputs=[content_output, summary_output, df_output, next_status]
         )
         
         send_btn.click(
-            fn=lambda: gr.update(visible=True),
+            fn=lambda: gr.update(value="Sending...", visible=True),
             inputs=[],
-            outputs=[send_loading]
+            outputs=[send_status]
         ).then(
-            fn=send_message_with_loading,
+            fn=send_message_with_status,
             inputs=[msg, chatbot, state],
-            outputs=[chatbot, msg, send_loading]
+            outputs=[chatbot, msg, send_status]
         )
         
         end_chat_btn.click(
-            fn=lambda: gr.update(visible=True),
+            fn=lambda: gr.update(value="Processing...", visible=True),
             inputs=[],
-            outputs=[end_chat_loading]
+            outputs=[end_chat_status]
         ).then(
-            fn=end_chat_with_loading,
+            fn=end_chat_with_status,
             inputs=[chatbot, state],
-            outputs=[gr.Textbox(label="Extracted Keywords"), results_page, end_chat_loading]
+            outputs=[gr.Textbox(label="Extracted Keywords"), results_page, end_chat_status]
         )
         
         export_btn.click(
-            fn=lambda: gr.update(visible=True),
+            fn=lambda: gr.update(value="Exporting...", visible=True),
             inputs=[],
-            outputs=[export_loading]
+            outputs=[export_status]
         ).then(
-            fn=export_with_loading,
+            fn=export_with_status,
             inputs=[results_df],
-            outputs=[gr.Textbox(label="Export Status"), export_loading]
+            outputs=[gr.Textbox(label="Export Status"), export_status]
         )
         
         restart_btn.click(