Update app.py

app.py

@@ -166,8 +166,6 @@ def mock_analytics():
         "Instagram": {"Views": random.randint(500, 3000), "Engagement Rate": f"{random.uniform(10, 20):.2f}%"},
     }
 
-import json
-
 def update_translations(file, edited_table):
     """
     Update the translations based on user edits in the Gradio Dataframe.
@@ -176,6 +174,8 @@ def update_translations(file, edited_table):
     logger.debug(f"Editable Table: {edited_table}")
 
     try:
+        start_time = time.time()  # Start the timer
+
         # Convert the edited_table (list of lists) back to list of dictionaries
         updated_translations = [
             {
@@ -190,52 +190,53 @@ def update_translations(file, 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
+        # Calculate elapsed time
+        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
 
     except Exception as e:
         raise ValueError(f"Error updating translations: {e}")
 
-# Core functionalities
 def upload_and_manage(file, language):
     if file is None:
-        return "Please upload a video/audio file.", None, None, None
+        return None, [], None, "No file uploaded. Please upload a video/audio file."
 
-    # Define paths for audio and output files
-    audio_path = "audio.wav"
-    output_video_path = "output_video.mp4"
+    try:
+        start_time = time.time()  # Start the timer
 
-    list_available_fonts()
-    # Transcribe audio from uploaded media file and get timestamps
-    transcrption_json = transcribe_video(file.name)
+        # Define paths for audio and output files
+        audio_path = "audio.wav"
+        output_video_path = "output_video.mp4"
 
-    translated_json = translate_text(transcrption_json, language)
-    
-    # Add transcript to video based on timestamps
-    add_transcript_to_video(file.name, translated_json, output_video_path)
+        list_available_fonts()
 
-    # Mock posting action (you can implement this as needed)
-    # post_message = mock_post_to_platform(platform, file.name)
+        # Transcribe audio from uploaded media file and get timestamps
+        transcrption_json = transcribe_video(file.name)
 
-    # Convert the translated JSON into a format for the editable table
-    editable_table = [
-        [float(entry["start"]), entry["original"], entry["translated"], float(entry["end"])]
-        for entry in translated_json
-    ]
+        # Translate the transcription
+        translated_json = translate_text(transcrption_json, language)
+        
+        # Add transcript to video based on timestamps
+        add_transcript_to_video(file.name, translated_json, output_video_path)
 
-    return translated_json, editable_table, output_video_path
+        # Convert the translated JSON into a format for the editable table
+        editable_table = [
+            [float(entry["start"]), entry["original"], entry["translated"], float(entry["end"])]
+            for entry in translated_json
+        ]
 
-# Wrapper to measure time elapsed
-def time_execution(func):
-    def wrapper(*args, **kwargs):
-        start_time = time.time()
-        result = func(*args, **kwargs)
+        # Calculate elapsed time
         elapsed_time = time.time() - start_time
-        return result, f"Time elapsed: {elapsed_time:.2f} seconds"
-    return wrapper
+        elapsed_time_display = f"Processing completed in {elapsed_time:.2f} seconds."
+
+        return translated_json, editable_table, output_video_path, elapsed_time_display
+
+    except Exception as e:
+        # Handle errors gracefully
+        return None, [], None, f"An error occurred: {str(e)}"
 
-# Wrap the functions with the timer
-timed_upload_and_manage = time_execution(upload_and_manage)
-timed_update_translations = time_execution(update_translations)
 
 # Gradio Interface with Tabs
 def build_interface():
@@ -284,13 +285,13 @@ def build_interface():
                     return message, gr.update(visible=False)            
 
             save_changes_button.click(
-                timed_update_translations, 
+                update_translations, 
                 inputs=[file_input, editable_table], 
                 outputs=[processed_video_output, elapsed_time_display]
             )
 
             submit_button.click(
-                timed_upload_and_manage, 
+                upload_and_manage, 
                 inputs=[file_input, language_input], 
                 outputs=[editable_translations, editable_table, processed_video_output, elapsed_time_display]
             )