Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 from openai import OpenAI
 import queue
 from datetime import datetime
+import time
 
 # Configuration for different API providers
 API_PROVIDERS = {
@@ -91,7 +92,7 @@ class TranslatorApp:
     def translate_with_chatgpt(self, text, source_lang, target_lang, api_key):
         client = OpenAI(api_key=api_key)
         response = client.chat.completions.create(
-            model="gpt-4o",
+            model="gpt-3.5-turbo",
             messages=[
                 {"role": "system", "content": f"Translate from {source_lang} to {target_lang}:"},
                 {"role": "user", "content": text},
@@ -124,9 +125,9 @@ class TranslatorApp:
             
             model = {
                 "DeepSeek": "deepseek-chat",
-                "ChatGPT": "gpt-4o",
+                "ChatGPT": "gpt-3.5-turbo",
                 "Mistral": "mistral-tiny"
-            }.get(api_provider, "gpt-4o")
+            }.get(api_provider, "gpt-3.5-turbo")
             
             response = client.chat.completions.create(
                 model=model,
@@ -143,12 +144,14 @@ class TranslatorApp:
     def log(self, message):
         self.log_queue.put(message)
 
-    def process_pptx(self, file_path, source_lang, target_lang, api_provider, api_key):
+    def process_pptx(self, file_path, source_lang, target_lang, api_provider, api_key, progress=gr.Progress()):
         try:
             prs = Presentation(file_path)
             total_slides = len(prs.slides)
             all_translated_text = ""
 
+            start_time = time.time()
+
             for i, slide in enumerate(prs.slides):
                 if not self.processing:
                     self.log("Process interrupted by user")
@@ -161,6 +164,10 @@ class TranslatorApp:
                         shape.text = translated_text
                         all_translated_text += translated_text + "\n"
 
+                # Update progress
+                elapsed_time = time.time() - start_time
+                progress((i + 1) / total_slides, desc=f"Processing slide {i + 1}/{total_slides}")
+
             if self.processing:
                 output_path = os.path.join(os.path.dirname(file_path), f"translated_{os.path.basename(file_path)}")
                 prs.save(output_path)
@@ -179,32 +186,46 @@ class TranslatorApp:
             self.log(f"Error during processing: {str(e)}")
             raise e
 
-    def run(self, file, api_provider, api_key, source_lang, target_lang):
+    def run(self, file, api_provider, api_key, source_lang, target_lang, progress=gr.Progress()):
         self.processing = True
         self.save_config(api_provider, api_key, source_lang, target_lang)
         try:
-            output_path, review_path = self.process_pptx(file.name, source_lang, target_lang, api_provider, api_key)
-            return f"File has been translated and saved as:\n{output_path}\nReview saved as:\n{review_path}"
+            output_path, review_path = self.process_pptx(file.name, source_lang, target_lang, api_provider, api_key, progress)
+            return [
+                f"File has been translated and saved as:\n{output_path}\nReview saved as:\n{review_path}",
+                output_path,
+                review_path
+            ]
         except Exception as e:
-            return f"An error occurred during processing: {str(e)}"
+            return [f"An error occurred during processing: {str(e)}", None, None]
         finally:
             self.processing = False
 
 def create_interface():
     app = TranslatorApp()
-    interface = gr.Interface(
-        fn=app.run,
-        inputs=[
-            gr.File(label="Upload PowerPoint File"),
-            gr.Dropdown(label="API Provider", choices=list(API_PROVIDERS.keys())),
-            gr.Textbox(label="API Key", type="password"),
-            gr.Dropdown(label="Source Language", choices=list(LANGUAGES.keys())),
-            gr.Dropdown(label="Target Language", choices=list(LANGUAGES.keys()))
-        ],
-        outputs=gr.Textbox(label="Output"),
-        title="BabelSlide by Heuristica",
-        description="Translate PowerPoint presentations using various AI models."
-    )
+    with gr.Blocks() as interface:
+        gr.Markdown("# BabelSlide by Heuristica")
+        gr.Markdown("Translate PowerPoint presentations using various AI models.")
+
+        with gr.Row():
+            file_input = gr.File(label="Upload PowerPoint File")
+            api_provider = gr.Dropdown(label="API Provider", choices=list(API_PROVIDERS.keys()))
+            api_key = gr.Textbox(label="API Key", type="password")
+            source_lang = gr.Dropdown(label="Source Language", choices=list(LANGUAGES.keys()))
+            target_lang = gr.Dropdown(label="Target Language", choices=list(LANGUAGES.keys()))
+
+        progress_bar = gr.Progress()
+        output_text = gr.Textbox(label="Output", interactive=False)
+        output_file = gr.File(label="Download Translated PowerPoint")
+        output_review = gr.File(label="Download Review")
+
+        run_button = gr.Button("Translate")
+        run_button.click(
+            fn=app.run,
+            inputs=[file_input, api_provider, api_key, source_lang, target_lang, progress_bar],
+            outputs=[output_text, output_file, output_review]
+        )
+
     return interface
 
 if __name__ == "__main__":