Update app.py

app.py

@@ -14,7 +14,7 @@ from groq import Groq
 from pydub import AudioSegment
 from moviepy.editor import AudioFileClip, concatenate_audioclips
 
-system_prompt = '''
+double_prompt = '''
     You are an insightful podcast generator. You have to create short conversations between Xiao and Yang that gives an overview of the News given by the user.
     Please provide the script and output strictly in the following JSON format:
     {
@@ -29,6 +29,21 @@ system_prompt = '''
     #Please note that the [string] you generate now must be in relaxed and natural Chinese.
 '''
 
+single_prompt = """
+    You are a podcast generator of sharp opinions and humor. You have to create short scripts for commentary on current events in the news given by the user.
+    Please provide the script and output strictly in the following JSON format:
+    {
+      "title": "[string]",
+      "content": {
+        "Yang_0: "[string]",
+        "Yang_0": "[string]",
+        ...
+      }
+    }
+    #Be concise. No less than five rounds of conversation.
+    #Please note that the [string] you generate now must be in relaxed and natural Chinese. 
+"""
+
 DESCRIPTION = '''
 <div>
 <h1 style="text-align: center;">📻听说demo</h1>
@@ -148,10 +163,10 @@ def extract_content(text):
     else:
         return None
 
-async def main(link):
+async def main(link, peoples="双人"):
     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:
@@ -159,6 +174,10 @@ async def main(link):
 
     prompt = f"News: {text}"
 
+    if peoples == "双人":
+        system_prompt = double_prompt;
+    else:
+        system_prompt = single_prompt;
     messages = [
         {"role": "system", "content": system_prompt},
         {"role": "user", "content": prompt},
@@ -212,6 +231,8 @@ with gr.Blocks(theme='soft', css=css, title="听说") as iface:
         output_box = gr.Audio(label="播客", type="filepath", interactive=False, autoplay=True, elem_classes="audio")  # Create an output textbox
     with gr.Row():
         input_box = gr.Textbox(label="网址", placeholder="请输入https开头的网址")
+    with gr.Row():
+        peoples = gr.CheckboxGrop(["单人","双人"],info="播音员人数")
     with gr.Row():
         submit_btn = gr.Button("🚀 发送")  # Create a submit button
         random_btn = gr.Button("🤙 随机")
@@ -219,7 +240,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, outputs=output_box)
+    submit_btn.click(main, inputs=[input_box, peoples], outputs=output_box)
     random_btn.click(fn=random_news, outputs=output_box)