Update app.py

app.py

@@ -14,8 +14,7 @@ from reportlab.lib.enums import TA_CENTER
 openai.api_key = os.environ['chat_key']
 
 
-def chat_gpt(user_message, model="gpt-3.5-turbo", max_tokens=4096):
-
+def trans_cn_en(user_message, model="gpt-3.5-turbo", max_tokens=4096):
     # 生成聊天回复
     response = openai.ChatCompletion.create(
         model= model,
@@ -34,9 +33,32 @@ def chat_gpt(user_message, model="gpt-3.5-turbo", max_tokens=4096):
     # 将助手回复添加到对应的会话中
     return assistant_response
 
+input_cn = gr.inputs.Textbox(lines=5, placeholder="请输入你的中文提示语")
+output_cn = gr.outputs.Textbox()
+iface_cn = gr.Interface(fn=trans_cn_en, inputs=input_cn, outputs=output_en, title="提示词", description="sd提示词中文转英文")
+
+def trans_en(user_message, model="gpt-3.5-turbo", max_tokens=4096):
+    # 生成聊天回复
+    response = openai.ChatCompletion.create(
+        model= model,
+        messages=[{"role": "system", "content": "translate my input into prompts using for stable diffusion in Chinese language"}] +
+                 [{"role": "user", "content": user_message}],
+        # system 中 定义回答问题的具体类型等
+        temperature=0.5,
+        max_tokens=150,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0,
+        stop=["\n\n"],
+    )
+    assistant_response = response.choices[0].message.content
+
+    # 将助手回复添加到对应的会话中
+    return assistant_response
 
-input_text = gr.inputs.Textbox(lines=5, placeholder="请输入你的中文提示语")
-output_text = gr.outputs.Textbox()
-iface = gr.Interface(fn=chat_gpt, inputs=input_text, outputs=output_text, title="提示词", description="sd提示词中文转英文")
+input_en = gr.inputs.Textbox(lines=5, placeholder="请输入你的英文提示语")
+output_en = gr.outputs.Textbox()
+iface_en = gr.Interface(fn=trans_en, inputs=input_en, outputs=output_en, title="提示词", description="sd提示词英文转中文")
 
-iface.launch()
+iface_cn.launch()
+iface_en.launch()