Update app.py

app.py

@@ -5,11 +5,10 @@ import time
 import threading
 import queue
 
-def generate_email_async(company, recipient_name, product, progress=gr.Progress()):
+def generate_email(company, recipient_name, product, progress=gr.Progress()):
     api_key = os.environ.get("BASETEN_API_KEY")
     if not api_key:
-        yield "Error: API_KEY not found in environment variables"
-        return
+        return "Error: API_KEY not found in environment variables"
 
     url = "https://model-4w56xy7q.api.baseten.co/development/predict"
     
@@ -24,8 +23,7 @@ def generate_email_async(company, recipient_name, product, progress=gr.Progress(
         "product_name": product
     }
 
-    yield "Initiating email generation..."
-    progress(0.1)
+    progress(0.1, "Initiating email generation...")
 
     def make_request(result_queue):
         try:
@@ -49,24 +47,22 @@ def generate_email_async(company, recipient_name, product, progress=gr.Progress(
     while thread.is_alive():
         elapsed_time = time.time() - start_time
         if elapsed_time > 300:  # 5 minutes timeout
-            yield "Error: Process took too long. Please try again later."
-            return
-        yield f"Still working... (Elapsed time: {elapsed_time:.0f} seconds)"
-        progress(min(0.1 + (elapsed_time / 300) * 0.8, 0.9))  # Progress up to 90%
+            return "Error: Process took too long. Please try again later."
+        progress(min(0.1 + (elapsed_time / 300) * 0.8, 0.9), f"Generating email... (Elapsed time: {elapsed_time:.0f} seconds)")
         time.sleep(5)  # Update every 5 seconds
 
     thread.join()
     status, result = result_queue.get()
 
     if status == "success":
-        progress(1.0)
-        yield f"Email generated successfully:\n\n{result}"
+        progress(1.0, "Email generated successfully!")
+        return result
     else:
-        yield result
+        return result
 
 # Create the Gradio interface
 iface = gr.Interface(
-    fn=generate_email_async,
+    fn=generate_email,
     inputs=[
         gr.Textbox(label="Company Name"),
         gr.Textbox(label="Recipient Name"),
@@ -75,8 +71,7 @@ iface = gr.Interface(
     outputs=gr.Textbox(label="Generated Email"),
     title="AI Email Generator",
     description="Generate personalized emails using AI (This process may take several minutes)",
-    allow_flagging="never",
-    live=True
+    allow_flagging="never"
 )
 
 # Launch the interface