update sth. at 2024-11-15 19:19:09

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import subprocess
 from huggingface_hub import InferenceClient
 from PIL import Image
 
@@ -33,6 +34,39 @@ def chat_with_model(messages):
         print(f"Chat generation failed: {e}")
         return "聊天生成失败，请稍后再试。"
 
+# ---------- chatgpt-4o-mini 模块 ----------
+def chatgpt_4o_mini(messages):
+    """
+    调用 gpt-4o-mini API 进行对话。
+    """
+    try:
+        # 构建请求数据
+        data = {
+            "model": "gpt-4o-mini",
+            "messages": [{"role": "system", "content": "你是一个辅助机器人"}] + messages,
+            "stream": True
+        }
+
+        # 执行 curl 请求
+        curl_command = [
+            "curl", "--location", "https://sanbo1200-duck2api.hf.space/completions",
+            "--header", "Content-Type: application/json",
+            "--data", str(data)
+        ]
+
+        # 使用 subprocess 调用 curl 命令
+        result = subprocess.run(curl_command, capture_output=True, text=True)
+        response = result.stdout
+
+        if result.returncode != 0:
+            return "请求失败，请稍后再试。"
+
+        # 解析返回结果
+        return response  # 这里你可以进一步解析 `response` 格式化成适当的返回信息
+    except Exception as e:
+        print(f"Error during gpt-4o-mini request: {e}")
+        return "gpt-4o-mini 请求失败，请稍后再试。"
+
 # ---------- 图像生成模块 ----------
 def image_gen(prompt):
     """
@@ -60,7 +94,7 @@ def image_gen(prompt):
 
 def build_interface():
     """
-    构建 Gradio 界面布局，包括文本聊天和图像生成模块。
+    构建 Gradio 界面布局，包括文本聊天、chatgpt-4o-mini 和图像生成模块。
     """
     with gr.Blocks() as demo:
         # 服务状态显示区域
@@ -78,6 +112,18 @@ def build_interface():
 
             chatbox_button.click(chat_handler, inputs=chatbox_input, outputs=chatbox_output)
 
+        # chatgpt-4o-mini 模块
+        with gr.Tab("chatgpt-4o-mini"):
+            chatgpt_input = gr.Textbox(label="输入你的问题", placeholder="请提问...")
+            chatgpt_output = gr.Textbox(label="回答")
+            chatgpt_button = gr.Button("发送")
+
+            def chatgpt_handler(user_input):
+                messages = [{"role": "user", "content": user_input}]
+                return chatgpt_4o_mini(messages)
+
+            chatgpt_button.click(chatgpt_handler, inputs=chatgpt_input, outputs=chatgpt_output)
+
         # 图像生成模块
         with gr.Tab("图像生成"):
             image_prompt = gr.Textbox(label="图像提示词", placeholder="描述你想生成的图像")
@@ -97,7 +143,7 @@ def build_interface():
             image_button.click(image_handler, inputs=image_prompt, outputs=[image_output_1, image_output_2])
 
         gr.Markdown("### 使用说明")
-        gr.Markdown("本助手支持文本聊天和图像生成功能，使用上方选项卡切换不同功能。")
+        gr.Markdown("本助手支持文本聊天、chatgpt-4o-mini 和图像生成功能，使用上方选项卡切换不同功能。")
 
     return demo