Add time tracking

app.py

@@ -1,5 +1,6 @@
 import io
 import os
+import time
 import streamlit as st
 import requests
 import zipfile
@@ -100,6 +101,8 @@ if uploaded_files and submit:
         # Add progress bar for translation status
         progress_bar = st.progress(0)
         for idx, uploaded_file in enumerate(uploaded_files):
+            # Start timing
+            start_time = time.time()  
 
             # Check file extension to determine translation method
             if uploaded_file.name.split('.')[-1] in ['txt', 'tsv', 'tab', 'csv', 'html', 'htm', 'mthml', 'mht', 'pptx', 'xlsx', 'docx', 'msg', 'xlf', 'xliff']:
@@ -107,13 +110,16 @@ if uploaded_files and submit:
             elif uploaded_file.name.split('.')[-1] in ['pdf', 'odt', 'odp', 'ods', 'rtf']:
                 result, response = process_async(uploaded_file)
 
+            # Calculate duration
+            duration = time.time() - start_time  
+
             # Check if translation was successful
             if result:
                 # Add successfully translated file to zip archive
                 zip_file.writestr(f"{lang_name}-translated-{uploaded_file.name}", response)
-                st.success(f"Successfully translated: {uploaded_file.name}")
+                st.success(f"Successfully translated: {uploaded_file.name} (Time taken: {duration:.2f} seconds)")
             else:
-                st.error(f"Failed to translate {uploaded_file.name} with status code {response.status_code}: {response.text}")
+                st.error(f"Failed to translate {uploaded_file.name} with status code {response.status_code}: {response.text} (Time taken: {duration:.2f} seconds)")
 
             # Update progress bar based on completed translations
             progress = (idx + 1) / len(uploaded_files)