WIP

app.py

@@ -138,7 +138,13 @@ mf_transcribe = gr.Interface(
         f" checkpoint [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe audio files"
         " of arbitrary length."
     ),
-    flagging_mode="manual"
+    flagging_mode="manual",
+    flagging_options=[
+        "Text Issue",
+        "Timestamp Issue",
+        "Missing Content",
+        "Other Issue"
+    ]
 )
 
 file_transcribe = gr.Interface(
@@ -160,91 +166,18 @@ file_transcribe = gr.Interface(
         f" checkpoint [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe audio files"
         " of arbitrary length."
     ),
-    flagging_mode="manual"
+    flagging_mode="manual",
+    flagging_options=[
+        "Text Issue",
+        "Timestamp Issue",
+        "Missing Content",
+        "Other Issue"
+    ]
 )
 
 # Then set up the demo with the interfaces
 with demo:
-    with gr.TabbedInterface([mf_transcribe, file_transcribe], ["Microphone", "Audio file"]) as tabs:
-        with gr.Row():
-            with gr.Column():
-                # Quick feedback
-                feedback_rating = gr.Radio(
-                    choices=["👍 Good", "👎 Bad"],
-                    label="Was this transcription accurate?",
-                    value="👍 Good"
-                )
-                
-                # Detailed feedback
-                with gr.Accordion("Detailed Feedback", open=False):
-                    flag_type = gr.Radio(
-                        choices=[
-                            "Text Issue",
-                            "Timestamp Issue",
-                            "Missing Content",
-                            "Other Issue"
-                        ],
-                        label="What type of issue did you find?",
-                        value="Text Issue"
-                    )
-                    
-                    # Correction submission
-                    with gr.Row():
-                        with gr.Column():
-                            gr.Markdown("### Original")
-                            original_text = gr.Textbox(
-                                label="Original text",
-                                interactive=False,
-                                lines=2
-                            )
-                        with gr.Column():
-                            gr.Markdown("### Correction")
-                            corrected_text = gr.Textbox(
-                                label="Corrected text",
-                                placeholder="Enter the correct text here",
-                                lines=2
-                            )
-                    
-                    # Timestamp correction
-                    with gr.Row():
-                        with gr.Column():
-                            gr.Markdown("### Original Timestamp")
-                            original_timestamp = gr.Textbox(
-                                label="Original timestamp",
-                                interactive=False,
-                                lines=1
-                            )
-                        with gr.Column():
-                            gr.Markdown("### Corrected Timestamp")
-                            corrected_timestamp = gr.Textbox(
-                                label="Corrected timestamp (HH:MM:SS,mmm)",
-                                placeholder="00:00:00,000",
-                                lines=1
-                            )
-                    
-                    flag_details = gr.Textbox(
-                        label="Additional notes",
-                        placeholder="Any other details about the issue...",
-                        lines=3
-                    )
-                
-                flag_button = gr.Button("Submit Feedback")
-            
-            # Setup flagging callback with all feedback components
-            flagging_callback.setup(
-                [tabs, feedback_rating, flag_type, original_text, corrected_text, 
-                 original_timestamp, corrected_timestamp, flag_details],
-                "flagged_data"
-            )
-            
-            # Handle flag submission
-            flag_button.click(
-                lambda *args: flagging_callback.flag(list(args)),
-                [tabs, feedback_rating, flag_type, original_text, corrected_text, 
-                 original_timestamp, corrected_timestamp, flag_details],
-                None,
-                preprocess=False
-            )
+    gr.TabbedInterface([mf_transcribe, file_transcribe], ["Microphone", "Audio file"])
 
 logger.info("Starting Gradio interface")
 demo.queue().launch(ssr_mode=False)