Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from io import BytesIO
 import fitz
-
+import tempfile
 import openai
 
 
@@ -68,35 +68,45 @@ def split_text(text, chunk_size=100):
 
 def translate_pdf(openai_key, pdf, start, stop):
     translator = TranslationAgent(openai_key)
-
-    # extract text
-    if pdf is not None:
-        text = extract_text_from_pdf(pdf, start=start, stop=stop)
-        chunks = split_text(text)
-
-        translated_chunks = []
-        for chunk in chunks:
-            translated_chunk = translator.translate_chunk(chunk)
-            translated_chunks.append(translated_chunk + " ")
-
-        translated_text = ' '.join(translated_chunks)
-        with open('translated.txt', 'w') as f:
-            f.write(translated_text)
-
-        return translated_text, "Translation Successful"
-
-
-
-iface = gr.Interface(title="Pdf Translator English -> Italian",
-                     fn=translate_pdf,
-                     inputs=[
-                         gr.inputs.Textbox(lines=1, label="OpenAI API key",
-                                           placeholder="Enter your OpenAI API key here"),
-                         gr.inputs.File(type="binary", label="PDF file", ),
-                         gr.inputs.Number(label="Starting Page", ),
-                         gr.inputs.Number(label="Final Page")
-                     ],
-                     outputs=["text", "text"]
-                     )
+    translated_text = ""
+    error_message = "Translation Successful"
+
+    try:
+        # extract text
+        if pdf is not None:
+            text = extract_text_from_pdf(pdf, start=start, stop=stop)
+            chunks = split_text(text)
+
+            translated_chunks = []
+            for chunk in chunks:
+                translated_chunk = translator.translate_chunk(chunk)
+                translated_chunks.append(translated_chunk + " ")
+
+            translated_text = ' '.join(translated_chunks)
+    except Exception as e:
+        error_message = f"Translation Failed: {e}"
+
+    # Create a temporary file with a specific prefix
+    temp = tempfile.NamedTemporaryFile(delete=False, prefix="translatedPDF_", suffix=".txt")
+
+    # Write to the temporary file
+    with open(temp.name, 'w', encoding='utf-8') as f:
+        f.write(translated_text)
+
+    return translated_text, error_message, temp.name
+
+
+iface = gr.Interface(
+    fn=translate_pdf,
+    inputs=[
+        gr.Textbox(lines=1, label="OpenAI API key",
+                          placeholder="Enter your OpenAI API key here"),
+        gr.File(type="binary", label="PDF file", ),
+        gr.Number(label="Starting Page", ),
+        gr.Number(label="Final Page")
+    ],
+    outputs=["text", "text", gr.File(label="Translated Text File")],
+    title="Pdf Translator: English ==> Italian",
+)
 
 iface.launch()