Update app.py

app.py

@@ -245,7 +245,9 @@ def create_ui() -> gr.Blocks:
                         lines=5,
                         placeholder="repo1, repo2\nrepo3"
                     )
-                    submit_btn = gr.Button("Submit Repo IDs", variant="primary")
+                    with gr.Row():
+                        submit_btn = gr.Button("Submit Repo IDs", variant="primary")
+                        submit_loading = gr.Loading(visible=False)
                     
                 with gr.Column():
                     gr.Markdown("### Or Search by Keywords")
@@ -254,7 +256,9 @@ def create_ui() -> gr.Blocks:
                         lines=3,
                         placeholder="Enter keywords separated by commas"
                     )
-                    search_btn = gr.Button("Search by Keywords", variant="primary")
+                    with gr.Row():
+                        search_btn = gr.Button("Search by Keywords", variant="primary")
+                        search_loading = gr.Loading(visible=False)
                     
             df_output = gr.Dataframe(
                 headers=["repo id", "strength", "weaknesses", "speciality", "relevance rating"],
@@ -262,6 +266,7 @@ def create_ui() -> gr.Blocks:
             )
             with gr.Row():
                 analyze_btn = gr.Button("Start Analysis", variant="primary")
+                analyze_loading = gr.Loading(visible=False)
                 compare_btn = gr.Button("Compare Repositories", variant="secondary")
             
         # Analysis Page
@@ -281,8 +286,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)
                 finish_btn = gr.Button("Finish Analysis", variant="secondary")
                 export_btn = gr.Button("Export Results", variant="secondary")
+                export_loading = gr.Loading(visible=False)
             
         # Comparison Page
         with gr.Group(visible=False) as comparison_page:
@@ -329,7 +336,9 @@ def create_ui() -> gr.Blocks:
             )
             with gr.Row():
                 send_btn = gr.Button("Send", variant="primary")
+                send_loading = gr.Loading(visible=False)
                 end_chat_btn = gr.Button("End Chat", variant="secondary")
+                end_chat_loading = gr.Loading(visible=False)
                 
         # Results Page
         with gr.Group(visible=False) as results_page:
@@ -385,15 +394,23 @@ def create_ui() -> gr.Blocks:
         )
         
         submit_btn.click(
+            fn=lambda: gr.update(visible=True),
+            inputs=[],
+            outputs=[submit_loading]
+        ).then(
             fn=process_repo_input,
             inputs=[repo_id_input, state],
-            outputs=[df_output]
+            outputs=[df_output, submit_loading]
         )
         
         search_btn.click(
+            fn=lambda: gr.update(visible=True),
+            inputs=[],
+            outputs=[search_loading]
+        ).then(
             fn=keyword_search_and_update,
             inputs=[keyword_input, state],
-            outputs=[df_output]
+            outputs=[df_output, search_loading]
         )
         
         analyze_btn.click(
@@ -403,9 +420,13 @@ def create_ui() -> gr.Blocks:
         )
         
         next_btn.click(
+            fn=lambda: gr.update(visible=True),
+            inputs=[],
+            outputs=[next_loading]
+        ).then(
             fn=show_combined_repo_and_llm,
             inputs=[state],
-            outputs=[content_output, summary_output, df_output]
+            outputs=[content_output, summary_output, df_output, next_loading]
         )
         
         finish_btn.click(
@@ -432,15 +453,23 @@ def create_ui() -> gr.Blocks:
             return keywords, navigate_to("results")
             
         send_btn.click(
-            fn=user_send,
+            fn=lambda: gr.update(visible=True),
+            inputs=[],
+            outputs=[send_loading]
+        ).then(
+            fn=send_message_with_loading,
             inputs=[msg, chatbot, state],
-            outputs=[chatbot, msg]
+            outputs=[chatbot, msg, send_loading]
         )
         
         end_chat_btn.click(
-            fn=end_chat,
+            fn=lambda: gr.update(visible=True),
+            inputs=[],
+            outputs=[end_chat_loading]
+        ).then(
+            fn=end_chat_with_loading,
             inputs=[chatbot, state],
-            outputs=[gr.Textbox(label="Extracted Keywords"), [start_page, input_page, analysis_page, chatbot_page, results_page]]
+            outputs=[gr.Textbox(label="Extracted Keywords"), [start_page, input_page, analysis_page, chatbot_page, results_page, help_page, comparison_page, history_page], end_chat_loading]
         )
         
         restart_btn.click(
@@ -486,9 +515,13 @@ def create_ui() -> gr.Blocks:
                 
         # Add new event handlers for new features
         export_btn.click(
-            fn=export_results,
+            fn=lambda: gr.update(visible=True),
+            inputs=[],
+            outputs=[export_loading]
+        ).then(
+            fn=export_with_loading,
             inputs=[results_df],
-            outputs=[gr.Textbox(label="Export Status")]
+            outputs=[gr.Textbox(label="Export Status"), export_loading]
         )
         
         history_btn.click(