Update app.py

app.py

@@ -72,13 +72,6 @@ rss_feed = 'https://www.scmp.com/rss/4/feed'
 apikey = os.environ.get("API_KEY")
 co = cohere.Client(api_key=apikey)
 
-system_prompt = ""
-
-def peoplesSelect(peoples): 
-    if peoples == "双人":
-        system_prompt = double_prompt
-    else:
-        system_prompt = single_prompt
 
 # RSS feeds
 def is_url(string):
@@ -178,8 +171,11 @@ async def main(link, peoples):
 
     prompt = f"Info: {text}"
 
-    peoplesSelect(peoples)
-    
+    if peoples == "双人":
+        system_prompt = double_prompt
+    else:
+        system_prompt = single_prompt
+
     messages = system_prompt + "\n\n\n" + prompt
     
     completion = co.chat(
@@ -207,7 +203,6 @@ async def main(link, peoples):
 
 
 async def random_news(peoples):
-    peoplesSelect(peoples)
     global rss_feed
     if not is_url(rss_feed):
         raise ValueError(f"{rss_feed} is not a valid RSS feed.")
@@ -217,7 +212,7 @@ async def random_news(peoples):
         news.append(entry.link)
     random_url = random.choice(news)
     print(random_url)
-    output = await main(random_url)
+    output = await main(random_url, peoples)
     return output
 
 Examples = [