Change to Docker

app.py

@@ -1,37 +1,31 @@
-import gradio as gr
+import os
+from fastapi import FastAPI
+from pydantic import BaseModel
+from fastapi.middleware.wsgi import WSGIMiddleware
 
-# 定义处理函数
-def api_aaa(text):
-    return text + 'aaa'
+app = FastAPI()  # 创建 FastAPI 应用
 
-def api_bbb(text):
-    return text + 'bbb'
+# 定义请求模型
+class TextRequest(BaseModel):
+    text: str
 
-# 创建两个独立的接口，分别对应两个功能
-iface_aaa = gr.Interface(
-    fn=api_aaa,
-    inputs="text",
-    outputs="text",
-    description="API endpoint for appending 'aaa' to text"
-)
+# 定义两个 API 路由处理函数
+@app.post("/api/aaa")
+async def api_aaa(request: TextRequest):
+    result = request.text + 'aaa'
+    return {"result": result}
 
-iface_bbb = gr.Interface(
-    fn=api_bbb,
-    inputs="text",
-    outputs="text",
-    description="API endpoint for appending 'bbb' to text"
-)
+@app.post("/api/bbb")
+async def api_bbb(request: TextRequest):
+    result = request.text + 'bbb'
+    return {"result": result}
 
-# 组合成 Blocks 页面
-with gr.Blocks() as demo:
-    gr.Markdown("# 模拟 API 接口")
-    
-    with gr.Tab("API /api/aaa"):
-        iface_aaa.render()
-        
-    with gr.Tab("API /api/bbb"):
-        iface_bbb.render()
+# Gradio 假界面，仅用于通过 Hugging Face Spaces 部署
+def fake_interface():
+    return "Gradio Interface Placeholder"
 
-# 启动 Gradio 应用并启用 API 访问
+# 启动应用，使用环境变量指定的端口
 if __name__ == "__main__":
-    demo.launch(enable_api=True)
+    import uvicorn
+    port = int(os.getenv("PORT", 7860))  # 获取 PORT 环境变量
+    uvicorn.run(app, host="0.0.0.0", port=port)