Update app.py

app.py

@@ -1,14 +1,11 @@
 import os
-import gradio as gr
 import json
-from tencentcloud.common import credential
-from tencentcloud.common.profile.client_profile import ClientProfile
-from tencentcloud.common.profile.http_profile import HttpProfile
-from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
-from tencentcloud.hunyuan.v20230901 import hunyuan_client, models
-
 from datetime import datetime
 
+import gradio as gr
+from openai import OpenAI
+
+
 def print_now(msg):
     now = datetime.now()
     formatted_time = now.strftime("%Y-%m-%d %H:%M:%S.%f")
@@ -27,38 +24,28 @@ def respond(
         default_system ="You are Tencent's helpful AI assistant Hunyuan."
 
         messages = [{"Role": "system", "Content": default_system}]
-
-        secret_id = os.getenv('SECRET_ID')
-        secret_key = os.getenv('SECRET_KEY')
-
-        cred = credential.Credential(secret_id, secret_key)
-        httpProfile = HttpProfile()
-        httpProfile.endpoint = "hunyuan.tencentcloudapi.com"
-        clientProfile= ClientProfile()
-        clientProfile.httpProfile = httpProfile
-        client = hunyuan_client.HunyuanClient(cred, "", clientProfile)
-        req = models.ChatCompletionsRequest()
-
+        client = OpenAI(
+            api_key=os.environ.get("HUNYUAN_API_KEY"), # 混元 APIKey
+            base_url="https://api.hunyuan.cloud.tencent.com/v1", # 混元 endpoint
+        )
         for val in history:
             if val[0] and val[1]:
                 messages.append({"Role": "user", "Content": val[0]})
                 messages.append({"Role": "assistant", "Content": val[1]})
         
         messages.append({"Role": "user", "Content": message})
-        params = {
-            "Model": "hunyuan-turbo",
-            "Messages": messages,
-            "Stream": True,
-            "StreamModeration": True,
-            "EnableEnhancement": False,
-        }
-        req.from_json_string(json.dumps(params))
-
-        resp= client.ChatCompletions(req)
-
+        completion = client.chat.completions.create(
+            model="hunyuan-turbo",
+            messages=messages,
+            stream=True,
+            extra_body={
+            "stream_moderation": True,
+            "enable_enhancement": False,
+            }
+        )
         response = ""
 
-        for event in resp:
+        for event in completion:
             data = json.loads(event['data'])
             if message.find('写一篇关于青春的五言绝句') != -1:
                 print(11111111111)