Add refresh functionality and auto-updates for history and metrics

app.py

@@ -281,7 +281,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                         value="python",
                         label="Language"
                     )
-                    submit_btn = gr.Button("Submit for Review")
+                    submit_btn = gr.Button("Submit for Review", variant="primary")
                 with gr.Column():
                     output = gr.Textbox(
                         label="Review Results",
@@ -289,39 +289,32 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                     )
         
         with gr.Tab("History"):
+            with gr.Row():
+                refresh_history = gr.Button("Refresh History", variant="secondary")
             history_output = gr.Textbox(
                 label="Review History",
                 lines=20,
-                value="No reviews yet."
+                value="Click 'Refresh History' to view review history"
             )
         
         with gr.Tab("Metrics"):
+            with gr.Row():
+                refresh_metrics = gr.Button("Refresh Metrics", variant="secondary")
             metrics_output = gr.JSON(
-                label="Performance Metrics",
-                value={
-                    'Total Reviews': 0,
-                    'Average Response Time': '0.00s',
-                    'Reviews Today': 0,
-                    'Device': 'Not initialized'
-                }
+                label="Performance Metrics"
             )
     
     @spaces.GPU
-    def review_code_interface(code: str, language: str) -> tuple:
+    def review_code_interface(code: str, language: str) -> str:
         if not code.strip():
-            return "Please enter some code to review.", None, None
+            return "Please enter some code to review."
         try:
             reviewer.ensure_initialized()
             result = reviewer.review_code(code, language)
-            
-            # Update history and metrics immediately
-            history = get_history_interface()
-            metrics = get_metrics_interface()
-            
-            return result, history, metrics
+            return result
         except Exception as e:
             logger.error(f"Interface error: {e}")
-            return f"Error: {str(e)}", None, None
+            return f"Error: {str(e)}"
     
     def get_history_interface() -> str:
         try:
@@ -343,16 +336,41 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
     
     def get_metrics_interface() -> Dict:
         try:
-            return reviewer.get_metrics()
+            metrics = reviewer.get_metrics()
+            if not metrics:
+                return {
+                    'Total Reviews': 0,
+                    'Average Response Time': '0.00s',
+                    'Reviews Today': 0,
+                    'Device': str(reviewer.device) if reviewer.device else "Not initialized"
+                }
+            return metrics
         except Exception as e:
             logger.error(f"Metrics error: {e}")
             return {"error": str(e)}
     
-    # Update all outputs when submitting code
+    # Set up event handlers
     submit_btn.click(
         review_code_interface,
         inputs=[code_input, language_input],
-        outputs=[output, history_output, metrics_output]
+        outputs=output
+    ).then(
+        get_history_interface,
+        outputs=history_output
+    ).then(
+        get_metrics_interface,
+        outputs=metrics_output
+    )
+    
+    # Add refresh buttons
+    refresh_history.click(
+        get_history_interface,
+        outputs=history_output
+    )
+    
+    refresh_metrics.click(
+        get_metrics_interface,
+        outputs=metrics_output
     )
     
     # Add example inputs