topic_output

app.py

@@ -3,17 +3,19 @@ import gradio as gr
 from openai import OpenAI
 
 
-
 OPEN_AI_KEY = os.getenv("OPEN_AI_KEY")
 OPEN_AI_CLIENT = OpenAI(api_key=OPEN_AI_KEY)
 
 
-
-
-def generate_topic_sentences(sys_content, user_content, model="gpt-4-1106-preview", max_tokens=4000):
+def generate_topic_sentences(model, max_tokens, sys_content,scenario, eng_level user_generate_topics_prompt):
     """
     根据系统提示和用户输入的情境及主题，调用OpenAI API生成相关的主题句。
     """
+    user_content = f"""
+        scenario is {scenario}
+        english level is {eng_level}
+        {user_generate_topics_prompt}
+    """
     messages = [
         {"role": "system", "content": sys_content},
         {"role": "user", "content": user_content}
@@ -30,28 +32,43 @@ def generate_topic_sentences(sys_content, user_content, model="gpt-4-1106-previe
 
     return content
 
+
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
+            # basic inputs
+            gr.Makdown("## Basic Inputs")
+            model = gr.Radio(["gpt-4-1106-preview", "gpt-3.5-turbo"], label="Model")
+            max_tokens = gr.Slider(minimum=50, maximum=4000, default=1000, label="Max Tokens")
+            sys_content_input = gr.Textbox(label="System Prompt", value="You are an English teacher who is practicing with me to improve my English writing skill.")
             scenario_input = gr.Textbox(label="Scenario")
-            topic_input = gr.Textbox(label="Topic")
             eng_level_input = gr.Radio(["beginner", "intermediate", "advanced"], label="English Level")
-            sys_content_input = gr.Textbox(label="System Prompt", value="You are an English teacher who is practicing with me to improve my English writing skill.")
-            user_content_template = "Give me 10 topics relevant to {}, for a paragraph. Just the topics, no explanation, use simple English language. Make sure the vocabulary you use is at {}."
+            
+            gr.Makdown("## Generate Topic Sentences")
+            user_generate_topics_prompt = "Give me 10 topics relevant to Scenario, for a paragraph. Just the topics, no explanation, use simple English language. Make sure the vocabulary you use is at english level."
             generate_topics_button = gr.Button("Generate Topic Sentences")
+
+            # gr.Makdown("## Generate Points")
+            # topic_input = gr.Textbox(label="Topic")
+            # generate_points_button = gr.Button("Generate Points")
+
         
         with gr.Column():
-            output = gr.Textbox(label="Generated Topic Sentences")
-    
-    # 当用户点击生成按钮时执行的操作
-    def on_generate_button_click(scenario, topic, eng_level, sys_content):
-        user_content = user_content_template.format(scenario, eng_level)
-        return generate_topic_sentences(sys_content, user_content)
+            topic_output = gr.Textbox(label="Generated Topic Sentences")
+            # points_output = gr.Textbox(label="Generated Points")
+
     
     generate_topics_button.click(
-        fn=on_generate_button_click,
-        inputs=[scenario_input, topic_input, eng_level_input, sys_content_input],
-        outputs=output
+        fn=generate_topic_sentences,
+        inputs=[
+            model, 
+            max_tokens, 
+            sys_content_input, 
+            scenario_input, 
+            eng_level_input,
+            user_generate_topics_prompt
+        ],
+        outputs=topic_output
     )
 
 demo.launch()