Fix history and metrics updates with proper state management

app.py

@@ -205,13 +205,30 @@ Code:
                 logger.error(f"Decoding error: {decode_error}")
                 return "Error: Failed to decode model output. Please try again."
             
+            # Create and save review
             end_time = datetime.now()
             review = Review(code, language, suggestions)
             review.response_time = (end_time - start_time).total_seconds()
+            
+            # Update metrics first
+            self.metrics['total_reviews'] += 1
+            total_time = self.metrics['avg_response_time'] * (self.metrics['total_reviews'] - 1)
+            total_time += review.response_time
+            self.metrics['avg_response_time'] = total_time / self.metrics['total_reviews']
+            
+            today = datetime.now().date()
+            
+            # Add review to history
             self.review_history.append(review)
             
-            self.update_metrics(review)
-            self.save_history()  # Save after each review
+            # Update today's reviews count
+            self.metrics['reviews_today'] = sum(
+                1 for r in self.review_history 
+                if datetime.fromisoformat(r.timestamp).date() == today
+            )
+            
+            # Save to file
+            self.save_history()
             
             if self.device and self.device.type == "cuda":
                 del inputs, outputs
@@ -349,20 +366,20 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
             logger.error(f"Metrics error: {e}")
             return {"error": str(e)}
     
-    # Set up event handlers
+    def update_all_outputs(code: str, language: str) -> tuple:
+        """Update all outputs after code review."""
+        result = review_code_interface(code, language)
+        history = get_history_interface()
+        metrics = get_metrics_interface()
+        return result, history, metrics
+    
+    # Connect the interface
     submit_btn.click(
-        review_code_interface,
+        update_all_outputs,
         inputs=[code_input, language_input],
-        outputs=output
-    ).then(
-        get_history_interface,
-        outputs=history_output
-    ).then(
-        get_metrics_interface,
-        outputs=metrics_output
+        outputs=[output, history_output, metrics_output]
     )
     
-    # Add refresh buttons
     refresh_history.click(
         get_history_interface,
         outputs=history_output