Update app.py

app.py

@@ -24,6 +24,9 @@ def create_interface():
             execution_time = gr.Textbox(label="Execution Time", visible=False)
             detected_language = gr.Textbox(label="Detected Language", visible=False)
             
+            # Add a Textbox for the SRT transcription
+            transcription_text = gr.Textbox(label="Transcription", visible=False, lines=10)
+            
             charts_and_explanations = []
             for _ in range(6):  # 3 analysis types * 2 speakers
                 with gr.Row():
@@ -33,16 +36,25 @@ def create_interface():
 
         def process_and_update(input_file):
             if input_file is None:
-                return [gr.update(value="No file selected")] + [gr.update(visible=False)] * 20
+                return [gr.update(value="No file selected")] + [gr.update(visible=False)] * 21  # +1 for transcription
             
             results = process_input(input_file, progress=gr.Progress())
             
             if len(results) != 10:
-                return [gr.update(value="Error: Unexpected number of results")] + [gr.update(visible=False)] * 20
+                return [gr.update(value="Error: Unexpected number of results")] + [gr.update(visible=False)] * 21
+            
+            # Extract transcription from results (assuming it's the last item)
+            transcription = results[-1]
+            
+            # Remove token information and transcription from results
+            results = results[:3] + results[3:6] + [transcription]
+            
+            visibility_updates = update_visibility_and_charts(*results[:6])
+            
+            # Add update for transcription text
+            transcription_update = gr.update(value=transcription, visible=True)
             
-            # Remove token information from results
-            results = results[:3] + results[3:6]
-            return update_visibility_and_charts(*results)
+            return visibility_updates + [transcription_update]
 
         def use_example_video():
             return example_video_path
@@ -50,7 +62,7 @@ def create_interface():
         input_file.upload(
             fn=process_and_update,
             inputs=[input_file],
-            outputs=[status_text, execution_time, detected_language] + charts_and_explanations
+            outputs=[status_text, execution_time, detected_language] + charts_and_explanations + [transcription_text]
         )
 
         example_button.click(
@@ -60,7 +72,7 @@ def create_interface():
         ).then(
             fn=process_and_update,
             inputs=[input_file],
-            outputs=[status_text, execution_time, detected_language] + charts_and_explanations
+            outputs=[status_text, execution_time, detected_language] + charts_and_explanations + [transcription_text]
         )
 
     return iface