feat: Simplify conversation verification UI for HF Spaces compatibility

src/interface/conversation_verification_ui.py

@@ -38,11 +38,10 @@ class VerificationInterface:
         self.current_session = session
         self.current_record_index = 0
 
-        with gr.Blocks(
-            title=f"Verify Conversation - {session.patient_name}"
-        ) as interface:
-            # Set theme after creation
-            interface.theme = gr.themes.Soft()
+        # Create interface without theme for HF Spaces compatibility
+        interface = gr.Blocks(title=f"Verify Conversation - {session.patient_name}")
+        
+        with interface:
             
             # Session header
             with gr.Row():
@@ -133,75 +132,54 @@ class VerificationInterface:
                         export_btn = gr.Button("📊 Export to CSV", variant="primary")
                         export_status = gr.HTML(value="")
 
-            # Hidden state components
-            session_state = gr.State(value=session)
-            current_index = gr.State(value=0)
-            
-            # Load initial exchange
-            interface.load(
-                fn=self._load_initial_exchange,
-                inputs=[session_state, current_index],
-                outputs=[exchange_display, current_position, stats_display, progress_bar]
-            )
+            # Initialize interface with first exchange data
+            if session.verification_records:
+                initial_exchange_html = self._render_exchange_review(session.verification_records[0])
+                initial_position_html = f"Exchange 1 of {len(session.verification_records)}"
+                initial_stats_html = self._render_statistics(session)
+                initial_progress_html = self._render_progress_bar(session)
+                
+                exchange_display.value = initial_exchange_html
+                current_position.value = initial_position_html
+                stats_display.value = initial_stats_html
+                progress_bar.value = initial_progress_html
+            else:
+                exchange_display.value = "No exchanges to verify"
+                current_position.value = "No exchanges"
+            
+            # Event handlers - simplified for HF Spaces compatibility
+            def handle_correct_click():
+                return self._handle_correct_feedback_simple()
             
-            # Event handlers
             correct_btn.click(
-                fn=self._handle_correct_feedback,
-                inputs=[session_state, current_index],
-                outputs=[exchange_display, current_position, stats_display, progress_bar, current_index]
+                fn=handle_correct_click,
+                outputs=[exchange_display, current_position, stats_display, progress_bar]
             )
             
+            def show_correction_section():
+                return gr.Column(visible=True)
+            
             incorrect_btn.click(
-                fn=lambda: gr.Column(visible=True),
+                fn=show_correction_section,
                 outputs=[correction_section]
             )
             
-            submit_correction_btn.click(
-                fn=self._handle_incorrect_feedback,
-                inputs=[session_state, current_index, correction_radio, correction_reason, correction_notes],
-                outputs=[exchange_display, current_position, stats_display, progress_bar, current_index, correction_section]
-            )
-            
-            cancel_correction_btn.click(
-                fn=lambda: (gr.Column(visible=False), "", "", ""),
-                outputs=[correction_section, correction_radio, correction_reason, correction_notes]
-            )
+            def handle_export_click():
+                return self._export_results_simple()
             
-            prev_btn.click(
-                fn=self._navigate_previous,
-                inputs=[session_state, current_index],
-                outputs=[exchange_display, current_position, current_index]
+            export_btn.click(
+                fn=handle_export_click,
+                outputs=[export_status]
             )
             
-            next_btn.click(
-                fn=self._navigate_next,
-                inputs=[session_state, current_index],
-                outputs=[exchange_display, current_position, current_index]
-            )
+            # Simplified handlers for basic functionality
+            def handle_mark_all_correct():
+                return self._mark_all_remaining_correct_simple()
             
             mark_all_correct_btn.click(
-                fn=self._mark_all_remaining_correct,
-                inputs=[session_state, current_index],
+                fn=handle_mark_all_correct,
                 outputs=[exchange_display, current_position, stats_display, progress_bar]
             )
-            
-            mark_green_correct_btn.click(
-                fn=self._mark_all_green_correct,
-                inputs=[session_state],
-                outputs=[exchange_display, current_position, stats_display, progress_bar, current_index]
-            )
-            
-            skip_to_errors_btn.click(
-                fn=self._skip_to_next_error,
-                inputs=[session_state, current_index],
-                outputs=[exchange_display, current_position, current_index]
-            )
-            
-            export_btn.click(
-                fn=self._export_results,
-                inputs=[session_state],
-                outputs=[export_status]
-            )
 
         return interface
 
@@ -674,4 +652,144 @@ class VerificationInterface:
             
         except Exception as e:
             print(f"Error in skip to errors: {e}")
-            return f"❌ Error: {str(e)}", "Error", current_index
+            return f"❌ Error: {str(e)}", "Error", current_index
+
+    def _handle_correct_feedback_simple(self) -> Tuple[str, str, str, str]:
+        """Simplified correct feedback handler for HF Spaces compatibility."""
+        try:
+            if not self.current_session or self.current_record_index >= len(self.current_session.verification_records):
+                return "No more exchanges", "Complete", "", ""
+            
+            record = self.current_session.verification_records[self.current_record_index]
+            
+            # Submit feedback
+            from src.core.conversation_verification import VerificationFeedback
+            feedback = VerificationFeedback(
+                exchange_id=record.exchange_id,
+                is_correct=True
+            )
+            
+            success = self.manager.submit_exchange_verification(self.current_session.session_id, record.exchange_id, feedback)
+            if not success:
+                return "❌ Failed to submit feedback", f"Exchange {self.current_record_index + 1}", "", ""
+            
+            # Reload session
+            updated_session = self.manager.load_session(self.current_session.session_id)
+            if not updated_session:
+                return "❌ Failed to reload session", f"Exchange {self.current_record_index + 1}", "", ""
+            
+            self.current_session = updated_session
+            
+            # Move to next unverified exchange
+            next_index = self._find_next_unverified_index(updated_session, self.current_record_index)
+            
+            if next_index is not None:
+                self.current_record_index = next_index
+                next_record = updated_session.verification_records[next_index]
+                exchange_html = self._render_exchange_review(next_record)
+                position_html = f"Exchange {next_index + 1} of {len(updated_session.verification_records)}"
+            else:
+                exchange_html = """
+                <div style='text-align: center; padding: 2em; background-color: #d4edda; border-radius: 8px;'>
+                    <h3>🎉 All exchanges verified!</h3>
+                    <p>Great job! You can now export the results.</p>
+                </div>
+                """
+                position_html = "Verification Complete"
+            
+            stats_html = self._render_statistics(updated_session)
+            progress_html = self._render_progress_bar(updated_session)
+            
+            return exchange_html, position_html, stats_html, progress_html
+            
+        except Exception as e:
+            print(f"Error in simplified correct feedback: {e}")
+            return f"❌ Error: {str(e)}", f"Exchange {self.current_record_index + 1}", "", ""
+
+    def _export_results_simple(self) -> str:
+        """Simplified export results for HF Spaces compatibility."""
+        try:
+            if not self.current_session:
+                return "❌ No session to export"
+            
+            verified_count = len([r for r in self.current_session.verification_records if r.is_correct is not None])
+            if verified_count == 0:
+                return """
+                <div style="background-color: #fff3cd; padding: 1em; border-radius: 4px;">
+                    <strong>⚠️ Nothing to Export</strong><br>
+                    Please verify some exchanges first.
+                </div>
+                """
+            
+            from src.core.verification_exporter import VerificationExporter
+            exporter = VerificationExporter()
+            csv_path = exporter.export_session_to_csv(self.current_session)
+            
+            import os
+            file_size = os.path.getsize(csv_path) if os.path.exists(csv_path) else 0
+            file_size_kb = file_size / 1024
+            
+            return f"""
+            <div style="background-color: #d4edda; padding: 1em; border-radius: 4px;">
+                <strong>✅ Export Successful!</strong><br>
+                <strong>File:</strong> <code>{os.path.basename(csv_path)}</code><br>
+                <strong>Size:</strong> {file_size_kb:.1f} KB<br>
+                <strong>Records:</strong> {verified_count} verified exchanges
+            </div>
+            """
+        except Exception as e:
+            return f"""
+            <div style="background-color: #f8d7da; padding: 1em; border-radius: 4px;">
+                <strong>❌ Export Failed</strong><br>
+                <strong>Error:</strong> {str(e)}
+            </div>
+            """
+
+    def _mark_all_remaining_correct_simple(self) -> Tuple[str, str, str, str]:
+        """Simplified mark all remaining correct for HF Spaces compatibility."""
+        try:
+            if not self.current_session:
+                return "❌ No session", "Error", "", ""
+            
+            unverified_records = self.current_session.get_unverified_records()
+            if not unverified_records:
+                return """
+                <div style='text-align: center; padding: 2em; background-color: #fff3cd; border-radius: 8px;'>
+                    <h3>⚠️ No unverified exchanges</h3>
+                    <p>All exchanges have already been verified.</p>
+                </div>
+                """, "All Verified", "", ""
+            
+            success_count = 0
+            from src.core.conversation_verification import VerificationFeedback
+            for record in unverified_records:
+                feedback = VerificationFeedback(
+                    exchange_id=record.exchange_id,
+                    is_correct=True
+                )
+                if self.manager.submit_exchange_verification(self.current_session.session_id, record.exchange_id, feedback):
+                    success_count += 1
+            
+            # Reload session
+            updated_session = self.manager.load_session(self.current_session.session_id)
+            if not updated_session:
+                return "❌ Failed to reload session", "Error", "", ""
+            
+            self.current_session = updated_session
+            
+            exchange_html = f"""
+            <div style='text-align: center; padding: 2em; background-color: #d4edda; border-radius: 8px;'>
+                <h3>🎉 Batch verification complete!</h3>
+                <p>Marked <strong>{success_count}</strong> exchanges as correct.</p>
+                <p>You can now export the results.</p>
+            </div>
+            """
+            position_html = "Verification Complete"
+            stats_html = self._render_statistics(updated_session)
+            progress_html = self._render_progress_bar(updated_session)
+            
+            return exchange_html, position_html, stats_html, progress_html
+            
+        except Exception as e:
+            print(f"Error in batch verification: {e}")
+            return f"❌ Error: {str(e)}", "Error", "", ""