Update app.py

app.py

@@ -54,7 +54,7 @@ model = AutoModelForCausalLM.from_pretrained(
 tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-1.8B-Chat")
 
 
-async def validate_url(url):
+def validate_url(url):
     try:
         response = httpx.get(url, timeout=60.0)
         response.raise_for_status()
@@ -66,7 +66,7 @@ async def validate_url(url):
     except Exception as e:
         return f"An unexpected error occurred: {str(e)}"
 
-async def fetch_text(url):
+def fetch_text(url):
     print("Entered Webpage Extraction")
     prefix_url = "https://r.jina.ai/"
     full_url = prefix_url + url
@@ -120,22 +120,7 @@ async def gen_show(script):
     return output_filename
 
 @spaces.GPU(duration=100)
-def generator(link):
-    if not link.startswith("http://") and not link.startswith("https://"):
-        return "URL must start with 'http://' or 'https://'",None
-
-    text = fetch_text(link)
-
-    if "Error" in text:
-        return text, None
-
-    prompt = f"News: {text}, json:"
-
-    messages = [
-        {"role": "system", "content": system_prompt},
-        {"role": "user", "content": prompt},
-    ]
-
+def generator(messages):
     answer = tokenizer.apply_chat_template(
         messages,
         tokenize=False,
@@ -153,8 +138,26 @@ def generator(link):
     ]
 
     results = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
+    return results
+    
 
-    generated_script = results
+async def main(link):
+    if not link.startswith("http://") and not link.startswith("https://"):
+        return "URL must start with 'http://' or 'https://'",None
+
+    text = fetch_text(link)
+
+    if "Error" in text:
+        return text, None
+
+    prompt = f"News: {text}, json:"
+
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": prompt},
+    ]
+    
+    generated_script = generator(messages)
 
     print("Generated Script:"+generated_script)
 
@@ -182,7 +185,7 @@ with gr.Blocks(theme='soft', css=css, title="Musen") as iface:
         clear_btn = gr.ClearButton(output_box, value="🗑️ Clear") # Create a clear button
 
     # Set up the event listeners
-    submit_btn.click(generator, inputs=input_box, outputs=output_box)
+    submit_btn.click(main, inputs=input_box, outputs=output_box)
 
 
 #gr.close_all()