Update app.py

app.py

@@ -166,12 +166,43 @@ def mock_analytics():
         "Instagram": {"Views": random.randint(500, 3000), "Engagement Rate": f"{random.uniform(10, 20):.2f}%"},
     }
 
-def update_translations(file, edited_translations):
+import json
+
+def update_translations(file, edited_table):
+    """
+    Update translations from the Gradio editable table and process the video.
+
+    Parameters:
+        file: Uploaded video file object.
+        edited_table: Updated translations from the Gradio editable table (list of dictionaries).
+
+    Returns:
+        output_video_path: The path to the video with updated translations.
+    """
     output_video_path = "output_video.mp4"
-    updated_translations = json.loads(edited_translations)
 
-    add_transcript_to_video(file.name, translated_json, output_video_path)
-    
+    try:
+        # Ensure edited_table is in the expected format
+        if not isinstance(edited_table, list):
+            raise ValueError("Edited table must be a list of dictionaries.")
+        
+        # Validate and preprocess each entry in the edited_table
+        for entry in edited_table:
+            if not all(key in entry for key in ["start", "original", "translated", "end"]):
+                raise ValueError("Each entry must contain 'start', 'original', 'translated', and 'end' keys.")
+            if not isinstance(entry["start"], (float, int)):
+                raise ValueError("'start' must be a number.")
+            if not isinstance(entry["end"], (float, int)):
+                raise ValueError("'end' must be a number.")
+            if not isinstance(entry["original"], str) or not isinstance(entry["translated"], str):
+                raise ValueError("'original' and 'translated' must be strings.")
+
+        # Call the function to add subtitles to the video
+        add_transcript_to_video(file.name, edited_table, output_video_path)
+
+    except Exception as e:
+        raise ValueError(f"Error updating translations: {e}")
+
     return output_video_path
 
 # Core functionalities
@@ -226,10 +257,19 @@ def build_interface():
             with gr.Column(scale=5):
 
                 gr.Markdown("## Edit Translations")
-                editable_translations = gr.Textbox(
-                    label="Edit Translated Texts (JSON Format)", 
-                    lines=10, 
-                    interactive=True
+                
+                # Editable JSON Data
+                editable_table = gr.Dataframe(
+                    value=[
+                        {"start": entry["start"], "original": entry["original"], "translated": entry["translated", "end": entry["end"]]}
+                        for entry in editable_translations
+                    ],
+                    headers=["start", "original", "translated", "end"],
+                    datatype=["number", "str", "str", "number"],
+                    row_count=len(translated_json),
+                    col_count=4,
+                    interactive=[False, True, True, False],  # Control editability
+                    label="Edit Translations",
                 )
                 save_changes_button = gr.Button("Save Changes")
                 processed_video_output = gr.File(label="Download Processed Video", interactive=True)  # Download button
@@ -254,7 +294,7 @@ def build_interface():
 
             save_changes_button.click(
                 update_translations, 
-                inputs=[file_input, editable_translations], 
+                inputs=[file_input, editable_table], 
                 outputs=[processed_video_output]
             )