Update app.py

app.py

@@ -478,29 +478,38 @@ def translate_text(transcription_json, source_language, target_language):
 
 def update_translations(file, edited_table, process_mode):
     """
-    Update the translations based on user edits in the Gradio Dataframe.
+    Update the translations based on user edits in the Gradio Dataframe
+    and allow the user to download the updated JSON.
     """
     output_video_path = "output_video.mp4"
+    updated_json_path = "updated_translations.json" # Define the path for the updated JSON file
     logger.debug(f"Editable Table: {edited_table}")
 
     if file is None:
         logger.info("No file uploaded. Please upload a video/audio file.")
-        return None, [], None, "No file uploaded. Please upload a video/audio file."
-        
+        return None, None, None
+
     try:
         start_time = time.time()  # Start the timer
 
-        # Convert the edited_table (list of lists) back to list of dictionaries
+        # Convert the edited_table (pandas DataFrame) back to list of dictionaries
+        # Ensure column names match the original structure
         updated_translations = [
             {
-                "start": row["start"],  # Access by column name
+                "start": row["start"],
                 "original": row["original"],
                 "translated": row["translated"],
-                "end": row["end"]
+                "end": row["end"],
+                "speaker": row["speaker"] # Include speaker if it's part of your translation structure
             }
             for _, row in edited_table.iterrows()
         ]
 
+        # Save the updated translations to a JSON file
+        with open(updated_json_path, 'w', encoding='utf-8') as f:
+            json.dump(updated_translations, f, indent=4, ensure_ascii=False)
+        logger.info(f"Updated translations saved to: {updated_json_path}")
+
         # Call the function to process the video with updated translations
         add_transcript_voiceover(file.name, updated_translations, output_video_path, process_mode)
 
@@ -508,10 +517,12 @@ def update_translations(file, edited_table, process_mode):
         elapsed_time = time.time() - start_time
         elapsed_time_display = f"Updates applied successfully in {elapsed_time:.2f} seconds."
 
-        return output_video_path, elapsed_time_display
-
+        # Return the path to the updated JSON file as well
+        return output_video_path, updated_json_path, elapsed_time_display
     except Exception as e:
-        raise ValueError(f"Error updating translations: {e}")
+        logger.error(f"Error updating translations: {e}")
+        # Return Nones for all outputs if an error occurs
+        return None, None, None
 
 def create_subtitle_clip_pil(text, start_time, end_time, video_width, video_height, font_path):
     try:
@@ -1336,13 +1347,16 @@ def build_interface():
         with gr.Row():
             with gr.Column(scale=4):
                 file_input = gr.File(label="Upload Video/Audio File")
-                language_input = gr.Dropdown(["en", "es", "fr", "zh"], label="Select Language")  # Language codes        
-                process_mode = gr.Radio(choices=[("Transcription Only", 1),("Transcription with Premium Voice",2),("Transcription with Voice Clone", 3)],label="Choose Processing Type",value=1)
-                submit_button = gr.Button("Post and Process")
+                language_input = gr.Dropdown(["en", "es", "fr", "zh"], label="Select Language")  # Language codes
 
+                process_mode = gr.Radio(choices=[("Transcription Only", 1),
+                                                 ("Transcription with Premium Voice", 2),
+                                                 ("Transcription with Voice Clone", 3)],
+                                       label="Choose Processing Type", value=1)
+                submit_button = gr.Button("Post and Process")
             with gr.Column(scale=8):
                 gr.Markdown("## Edit Translations")
-                
+
                 # Editable JSON Data
                 editable_table = gr.Dataframe(
                     value=[],  # Default to an empty list to avoid undefined values
@@ -1358,7 +1372,6 @@ def build_interface():
                 processed_video_output = gr.File(label="Download Processed Video", interactive=True)  # Download button
                 elapsed_time_display = gr.Textbox(label="Elapsed Time", lines=1, interactive=False)
                 translated_json_download = gr.File(label="Download Translated JSON", interactive=True) # New: JSON download
-
             with gr.Column(scale=1):
                 gr.Markdown("**Feedback**")
                 feedback_input = gr.Textbox(
@@ -1369,33 +1382,30 @@ def build_interface():
                 feedback_btn = gr.Button("Submit Feedback")
                 response_message = gr.Textbox(label=None, lines=1, interactive=False)
                 db_download = gr.File(label="Download Database File", visible=False)
-            
-                # Link the feedback handling
-                def feedback_submission(feedback):
-                    message, file_path = handle_feedback(feedback)
-                    if file_path:
-                        return message, gr.update(value=file_path, visible=True)
-                    return message, gr.update(visible=False)            
+
+            # Link the feedback handling
+            def feedback_submission(feedback):
+                message, file_path = handle_feedback(feedback)
+                if file_path:
+                    return message, gr.update(value=file_path, visible=True)
+                return message, gr.update(visible=False)
 
             save_changes_button.click(
-                update_translations, 
+                update_translations,
                 inputs=[file_input, editable_table, process_mode],
-                outputs=[processed_video_output, elapsed_time_display]
+                outputs=[processed_video_output, translated_json_download, elapsed_time_display]
             )
-
             submit_button.click(
-                upload_and_manage, 
-                inputs=[file_input, language_input, process_mode], 
+                upload_and_manage,
+                inputs=[file_input, language_input, process_mode],
                 outputs=[editable_table, processed_video_output, translated_json_download, elapsed_time_display]
             )
-
             # Connect submit button to save_feedback_db function
             feedback_btn.click(
-                feedback_submission, 
-                inputs=[feedback_input], 
+                feedback_submission,
+                inputs=[feedback_input],
                 outputs=[response_message, db_download]
             )
-
     return demo
 
 tts_model = None