Update app.py

app.py

@@ -1,42 +1,44 @@
 import gradio as gr
-import requests
 import os
+from openai import OpenAI
 
-# 假设 DeepSeek-R1 的 API URL 和 API Key
-DEEPSEEK_API_URL = os.getenv("DEEPSEEK_API_URL", "https://dashscope.aliyuncs.com/compatible-mode/v1")
-DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
+# 读取环境变量
+DASHSCOPE_API_KEY = os.getenv("DASHSCOPE_API_KEY")  # 从环境变量读取 API Key
+DASHSCOPE_API_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
+
+# 设置 DashScope API 客户端
+client = OpenAI(
+    api_key=DASHSCOPE_API_KEY,
+    base_url=DASHSCOPE_API_URL
+)
 
 # 定义对话函数
 def answer(question, history):
     if history is None:
         history = []
 
-    # 组装对话历史
-    messages = [{"role": "system", "content": "你是一个中国厨师，用中文回答做菜的问题。"}]
-
-    for item in history:
-        messages.append({"role": item["role"], "content": item["content"]})
-
-    messages.append({"role": "user", "content": question})
-
-    # 调用 DeepSeek-R1 API
+    # 将历史记录与用户输入合并
+    messages = [{'role': 'user', 'content': question}]
+    
+    # 向 DashScope API 发送请求
     try:
-        # 发送请求到 DeepSeek-R1 API
-        response = requests.post(
-            DEEPSEEK_API_URL,
-            json={"messages": messages},
-            headers={"Authorization": f"Bearer {DEEPSEEK_API_KEY}"}
+        completion = client.chat.completions.create(
+            model="deepseek-r1",  # 使用 DeepSeek-R1 模型
+            messages=messages
         )
-        
-        # 解析 API 响应
-        response_data = response.json()
-        print("API 响应数据：", response_data)  # 打印响应数据进行调试
-        bot_response = response_data.get("choices")[0]["message"]["content"]
+
+        # 获取思考过程与最终答案
+        reasoning_content = completion.choices[0].message.get("reasoning_content", "没有思考过程")
+        bot_response = completion.choices[0].message["content"]
+
+        # 输出思考过程与最终答案（用于调试）
+        print("思考过程：", reasoning_content)
+        print("最终答案：", bot_response)
 
         # 更新历史记录
         history.append({"role": "user", "content": question})
         history.append({"role": "assistant", "content": bot_response})
-        
+
         return history, history
 
     except Exception as e:
@@ -105,4 +107,4 @@ with gr.Blocks(css="""
 
     txt.submit(answer, [txt, state], [chatbot, state])
 
-demo.launch()
+demo.launch()