Change to Docker

app.py

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