Update app.py

app.py

@@ -1,34 +1,35 @@
-from fastapi import FastAPI, Request
-import os, requests
+import gradio as gr
+import os
+import requests
 
-app = FastAPI()
-# 获取你的 Secret
-HF_TOKEN = os.environ.get('HF_TOKEN')
-# ✅ 核心：这是针对你这个模型的唯一合法地址
-API_URL = "https://router.huggingface.co/jiang1002/chatglm-6b-adgen"
+# 这里的逻辑是：你的 Space 已经是一个运行环境了
+# 我们直接在网页上通过 Gradio 搭建对话框
 
-@app.post("/generate")
-async def generate(request: Request):
-    try:
-        body = await request.json()
-        headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-        payload = {
-            "inputs": body.get("text", "你好"),
-            "parameters": {"max_new_tokens": 512, "do_sample": True},
-            "options": {"wait_for_model": True}
-        }
-        
-        # 发送请求
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=120)
-        
-        if response.status_code == 200:
-            return {"success": True, "response": response.json()}
-        else:
-            # 这里的详细报错能帮我们抓鬼
-            return {"success": False, "error": f"HF_{response.status_code}", "detail": response.text}
-    except Exception as e:
-        return {"success": False, "error": "SPACE_INTERNAL_ERROR", "detail": str(e)}
+def chat_function(message, history):
+    # 这里依然调用你本地模型或 API 的逻辑，但我们换一种更稳的方式
+    # 如果你本地有加载模型代码，就写在这里；
+    # 如果还是想调 API，我们用这个最简单的逻辑：
+    
+    API_URL = "https://router.huggingface.co/jiang1002/chatglm-6b-adgen"
+    headers = {"Authorization": f"Bearer {os.environ.get('HF_TOKEN')}"}
+    payload = {"inputs": message, "options": {"wait_for_model": True}}
+    
+    response = requests.post(API_URL, headers=headers, json=payload)
+    
+    if response.status_code == 200:
+        result = response.json()
+        if isinstance(result, list):
+            return result[0].get("generated_text", "出错了")
+        return str(result)
+    else:
+        return f"错误：{response.status_code}, {response.text}"
 
-@app.get("/")
-def home():
-    return {"status": "Running", "token_set": bool(HF_TOKEN), "current_endpoint": API_URL}
+# 搭建漂亮的网页界面
+demo = gr.ChatInterface(
+    fn=chat_function, 
+    title="ChatGLM-6B 广告生成助手",
+    description="点击下方输入框，输入你的需求，直接开始测试！"
+)
+
+if __name__ == "__main__":
+    demo.launch()