Update app.py

app.py

@@ -18,7 +18,7 @@ from nltk.tokenize import sent_tokenize
 import logging
 from textblob import TextBlob
 import whisper
-
+import time
 import sqlite3
 
 # Define the passcode
@@ -224,35 +224,31 @@ def upload_and_manage(file, language):
 
     return translated_json, editable_table, output_video_path
 
-# def generate_dashboard():
-#     # Mock analytics generation
-#     analytics = mock_analytics()
-
-#     if not analytics:
-#         return "No analytics available."
+# Wrapper to measure time elapsed
+def time_execution(func):
+    def wrapper(*args, **kwargs):
+        start_time = time.time()
+        result = func(*args, **kwargs)
+        elapsed_time = time.time() - start_time
+        return result, f"Time elapsed: {elapsed_time:.2f} seconds"
+    return wrapper
 
-#     dashboard = "Platform Analytics:\n"
-#     for platform, data in analytics.items():
-#         dashboard += f"\n{platform}:\n"
-#         for metric, value in data.items():
-#             dashboard += f"  {metric}: {value}\n"
-#     return dashboard
+# 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():
     with gr.Blocks() as demo:
-        # with gr.Tab("Content Management"):
         gr.Markdown("## Video Localization")
         with gr.Row():
             with gr.Column(scale=4):
                 file_input = gr.File(label="Upload Video/Audio File")
-                # platform_input = gr.Dropdown(["YouTube", "Instagram"], label="Select Platform")
                 language_input = gr.Dropdown(["en", "es", "fr", "zh"], label="Select Language")  # Language codes
                 submit_button = gr.Button("Post and Process")
                 editable_translations = gr.State(value=[])
 
             with gr.Column(scale=8):
-
                 gr.Markdown("## Edit Translations")
                 
                 # Editable JSON Data
@@ -267,6 +263,7 @@ def build_interface():
                 )
                 save_changes_button = gr.Button("Save Changes")
                 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)
 
             with gr.Column(scale=1):
                 gr.Markdown("**Feedback**")
@@ -275,7 +272,7 @@ def build_interface():
                     label=None,
                     lines=3,
                 )
-                feedback_btn = gr.Button("Submit Feeback")
+                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)
             
@@ -284,18 +281,18 @@ def build_interface():
                     message, file_path = handle_feedback(feedback)
                     if file_path:
                         return message, gr.update(value=file_path, visible=True)
-                    return message, gr.update(visible=False)
+                    return message, gr.update(visible=False)            
 
             save_changes_button.click(
-                update_translations, 
+                timed_update_translations, 
                 inputs=[file_input, editable_table], 
-                outputs=[processed_video_output]
+                outputs=[processed_video_output, elapsed_time_display]
             )
 
             submit_button.click(
-                upload_and_manage, 
+                timed_upload_and_manage, 
                 inputs=[file_input, language_input], 
-                outputs=[editable_translations, editable_table, processed_video_output]
+                outputs=[editable_translations, editable_table, processed_video_output, elapsed_time_display]
             )
         
             # Connect submit button to save_feedback_db function
@@ -305,13 +302,6 @@ def build_interface():
                 outputs=[response_message, db_download]
             )
 
-    # with gr.Tab("Analytics Dashboard"):
-    #     gr.Markdown("## Content Performance Analytics")
-    #     analytics_output = gr.Textbox(label="Dashboard", interactive=False)
-    #     generate_dashboard_button = gr.Button("Generate Dashboard")
-
-    #     generate_dashboard_button.click(generate_dashboard, outputs=[analytics_output])
-
     return demo
 
 # Launch the Gradio interface