Update app.py

app.py

@@ -4,6 +4,7 @@ import openai
 import gradio as gr
 import os
 from dotenv import load_dotenv
+import time
 
 # Load environment variables
 load_dotenv()
@@ -56,7 +57,7 @@ def create_newsletter(contents, language):
     response = openai.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
-            {"role": "system", "content": "You are a helpful assistant and an expert in making newsletters on AI for The Neural (www.theneural.ai)."},
+            {"role": "system", "content": "You are a helpful assistant expert in making newsletters."},
             {"role": "user", "content": prompt}
         ]
     )
@@ -68,13 +69,29 @@ def process_urls(url1, url2, url3, url4, url5, language):
     if not urls:
         return "No URLs provided."
     
+    with gr.update() as progress:
+        progress(0.2)
+        time.sleep(1)
+    
     contents = []
     for url in urls:
         title, content = scrape_content(url)
         contents.append((title, content, url))
     
+    with gr.update() as progress:
+        progress(0.6)
+        time.sleep(1)
+    
     newsletter = create_newsletter(contents, language)
-    return newsletter
+    with gr.update() as progress:
+        progress(1.0)
+        time.sleep(1)
+    
+    file_path = "newsletter.txt"
+    with open(file_path, "w", encoding="utf-8") as file:
+        file.write(newsletter)
+    
+    return newsletter, file_path
 
 # Gradio interface
 iface = gr.Interface(
@@ -87,9 +104,9 @@ iface = gr.Interface(
         gr.Textbox(label="URL 5"),
         gr.Dropdown(choices=list(LANGUAGES.keys()), label="Select Language", value="English")
     ],
-    outputs="html",
+    outputs=["html", gr.File(label="Download Newsletter")],
     title="Multilingual AI Newsletter Generator",
-    description="Enter up to 5 URLs to generate a newsletter in your preferred language. Copy and paste the output into your CMS tool for further editing."
+    description="Enter up to 5 URLs to generate a newsletter in your preferred language. Copy and paste the output into your CMS tool for further editing. A progress indicator will show the process, and you can download the newsletter as a text file."
 )
 
 iface.launch()