Update app.py

app.py

@@ -170,41 +170,30 @@ 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.
+    Update the translations based on user edits in the Gradio Dataframe.
     """
     output_video_path = "output_video.mp4"
 
     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)
+        # Convert the edited_table (list of lists) back to list of dictionaries
+        updated_translations = [
+            {
+                "start": row[0],  # First column
+                "original": row[1],  # Second column
+                "translated": row[2],  # Third column
+                "end": row[3],  # Fourth column
+            }
+            for row in edited_table
+        ]
+
+        # Call the function to process the video with updated translations
+        add_transcript_to_video(file.name, updated_translations, output_video_path)
+
+        return output_video_path
 
     except Exception as e:
         raise ValueError(f"Error updating translations: {e}")
 
-    return output_video_path
-
 # Core functionalities
 def upload_and_manage(file, language):
     if file is None: