51 lines
1.9 KiB
Python
51 lines
1.9 KiB
Python
import gradio as gr
|
||
import pandas as pd
|
||
import tempfile
|
||
import csv
|
||
from http import HTTPStatus
|
||
import dashscope
|
||
from dashscope import Generation
|
||
import os
|
||
|
||
dashscope.api_key = os.getenv("DASHSCOPE_API_KEY") # Vincent's API key
|
||
|
||
|
||
def response(prompt, instruction=None):
|
||
messages = [{'role': 'user', 'content': prompt}]
|
||
if instruction is not None: # 如果提供了指令，则添加到messages中
|
||
messages.insert(0, {'role': 'system', 'content': instruction})
|
||
|
||
response = Generation.call(model='qwen-plus',
|
||
messages=messages,
|
||
seed=1234,
|
||
result_format='message',
|
||
stream=False,
|
||
incremental_output=False,
|
||
temperature=1.8,
|
||
top_p=0.9,
|
||
top_k=999
|
||
)
|
||
if response.status_code == HTTPStatus.OK:
|
||
message = response.output.choices[0]['message']['content']
|
||
return message
|
||
else:
|
||
print('Request id: %s, Status code: %s, error code: %s, error message: %s' % (
|
||
response.request_id, response.status_code,
|
||
response.code, response.message
|
||
))
|
||
return f"Error: Could not generate response with Status code: {response.status_code}, error code: {response.code}"
|
||
|
||
|
||
with gr.Blocks() as demo:
|
||
input_text = gr.Textbox(label="输入文本")
|
||
output_text = gr.Textbox(label="输出文本")
|
||
submit_button = gr.Button("submit")
|
||
# 使用 gr.ClearButton 来清除指定的输出组件
|
||
clear_button = gr.ClearButton(components=[output_text], value="Clear processed data")
|
||
|
||
# 当输入文本发生变化时，调用 response 函数并将结果显示在输出文本框中
|
||
submit_button.click(fn=response, inputs=input_text, outputs=output_text)
|
||
|
||
demo.launch()
|
||
|