Update app.py

app.py

@@ -72,9 +72,9 @@ rss_feed = 'https://www.scmp.com/rss/4/feed'
 apikey = os.environ.get("API_KEY")
 co = cohere.Client(api_key=apikey)
 
+system_prompt = ""
 
 # RSS feeds
-
 def is_url(string):
     url_pattern = re.compile(
         r'^(?:http|ftp)s?://'  # http:// or https://
@@ -161,21 +161,17 @@ def extract_content(text):
     else:
         return None
 
-async def main(link, peoples="双人"):
+async def main(link):
+    global system_prompt
     if not link.startswith("http://") and not link.startswith("https://"):
         return "URL must start with 'http://' or 'https://'",None
-    system_prompt = ""
+    
     text = fetch_text(link)
 
     if "Error" in text:
         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
 
@@ -231,6 +227,10 @@ with gr.Blocks(theme='soft', css=css, title="听说") as iface:
         input_box = gr.Textbox(label="网址", placeholder="请输入https开头的网址")
     with gr.Row():
         peoples = gr.Radio(["单人","双人"],value="双人",label="播音员人数")
+        if peoples == "双人":
+            system_prompt = double_prompt;
+        else:
+            system_prompt = single_prompt;
     with gr.Row():
         submit_btn = gr.Button("🚀 发送")  # Create a submit button
         random_btn = gr.Button("🤙 随机")
@@ -238,7 +238,7 @@ 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")
         
     # Set up the event listeners
-    submit_btn.click(main, inputs=[input_box, peoples], outputs=output_box)
+    submit_btn.click(main, inputs=input_box, outputs=output_box)
     random_btn.click(fn=random_news, outputs=output_box)