Update app.py

app.py

@@ -72,7 +72,7 @@ rss_feed = 'https://www.scmp.com/rss/4/feed'
 apikey = os.environ.get("API_KEY")
 co = cohere.Client(api_key=apikey)
 
-system_prompt = ""
+peoples = ""
 
 # RSS feeds
 def is_url(string):
@@ -161,8 +161,8 @@ def extract_content(text):
     else:
         return None
 
-async def main(link):
-    global system_prompt
+async def main(link, peoples):
+    global peoples
     if not link.startswith("http://") and not link.startswith("https://"):
         return "URL must start with 'http://' or 'https://'",None
     
@@ -172,9 +172,13 @@ async def main(link):
         return text, None
 
     prompt = f"Info: {text}"
+
+    if peoples == "双人":
+        system_prompt = double_prompt
+    else:
+        system_prompt = single_prompt
     
     messages = system_prompt + "\n\n\n" + prompt
-
     
     completion = co.chat(
         model="command-r",
@@ -200,7 +204,8 @@ async def main(link):
     return output_filename
 
 
-async def random_news():
+async def random_news(peoples):
+    global peoples
     global rss_feed
     if not is_url(rss_feed):
         raise ValueError(f"{rss_feed} is not a valid RSS feed.")
@@ -234,13 +239,9 @@ with gr.Blocks(theme='soft', css=css, title="听说") as iface:
 
     gr.Examples(examples=Examples, inputs=input_box, outputs=output_box, fn=main, label="示例", cache_examples="lazy")
 
-    if peoples == "双人":
-        system_prompt = double_prompt
-    else:
-        system_prompt = single_prompt
     # Set up the event listeners
-    submit_btn.click(main, inputs=input_box, outputs=output_box)
-    random_btn.click(fn=random_news, outputs=output_box)
+    submit_btn.click(main, inputs=[input_box,peoples], outputs=output_box)
+    random_btn.click(fn=random_news, inputs=peoples, outputs=output_box)
 
 
 #gr.close_all()