Update app.py

app.py

@@ -1,6 +1,28 @@
 from openai import OpenAI
 import gradio as gr
 
+def generate_response(question, model_name, endpoint, api_key):
+    # 创建 OpenAI 客户端
+    client = OpenAI(api_key=api_key, base_url=endpoint)
+    messages = [
+        {"role": "user", "content": question}
+    ]
+
+    try:
+        response = client.chat.completions.create(
+            model=model_name,
+            messages=messages,
+            max_tokens=100,
+            temperature=0.1
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"响应生成失败: {str(e)}"
+
+base_url_options=["https://lmzh.top/v1","https://api.openai.com"]
+model_options = ["gpt-3.5-turbo", "gpt-4o-mini", "gpt-4o"]
+
+
 def validate_tran_api_key(api_key):
     if api_key == '':
         raise gr.Error('请输入您的中转API Key')
@@ -82,8 +104,16 @@ with app:
         with gr.Column():
             official_output_textbox = gr.Textbox(label="显示key状态及可访问的模型列表", lines=5, max_lines=5, elem_classes="custom-textbox" )
         official_btn_text.click(validate_offi_api_key, official_api_key, official_output_textbox)
-
-
+    with gr.Row(variant='panel'):
+        with gr.Column():
+            question=gr.Textbox(label="请输入你的问题: ")
+            model_name=gr.Dropdown(label="选择模型", choices=model_options, value="gpt-3.5-turbo")
+            endpoint=gr.Dropdown(label="选择接口地址", choices=base_url_options, value="https://lmzh.top/v1")
+            api_key=gr.Textbox(type='password', label='api-key', placeholder='请在此填写您的API Key')
+            message_btn = gr.Button("发送", variant="primary", elem_classes="custom-button")
+        with gr.Column():
+            output_textbox = gr.Textbox(label="显示问题答案", lines=5, max_lines=5, elem_classes="custom-textbox" )
+            message_btn.click(generate_response,inputs=[question,mode_name,endpoint,api_key],outputs=message_textbox) 
 #添加页面底部
     gr.HTML('''
         <div class="footer">