feat: update

app.py

@@ -1,45 +1,65 @@
 from typing import List, Literal
 from pydantic import BaseModel, Field
+import gradio as gr
 from fastapi import FastAPI, APIRouter, Request
 from fastapi.middleware.cors import CORSMiddleware
 from sentence_transformers import SentenceTransformer
 import uvicorn
+import requests
+import asyncio
+import threading
 
-# Initialize FastAPI app
+# Khởi tạo FastAPI
 app = FastAPI()
 
-# CORS middleware
+# Thêm middleware CORS để cho phép yêu cầu từ Gradio
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
+    allow_origins=["*"],  # Cho phép tất cả các nguồn
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# Load model
-model = SentenceTransformer('Alibaba-NLP/gte-multilingual-base', trust_remote_code=True)
+# Tải mô hình
+model = SentenceTransformer(model_name_or_path='Alibaba-NLP/gte-multilingual-base', trust_remote_code=True)
 
-# Define data model
+# Định nghĩa mô hình dữ liệu cho yêu cầu
 class PostEmbeddings(BaseModel):
     type: Literal['default', 'disease', 'gte'] = Field(default='default')
     sentences: List[str]
 
-# Router for embeddings
-router = APIRouter(prefix="/retrieval", tags=["retrieval"])
+# Tạo router cho API
+router = APIRouter(
+    prefix="/retrieval",
+    tags=["retrieval"],
+    responses={404: {"description": "Not found"}},
+)
 
-@router.post('/embeddings')
-def post_embeddings(request: Request, data: PostEmbeddings):
+@app.post("/retrieval/embeddings")
+def post_embeddings(data: PostEmbeddings):
     embeddings = model.encode(data.sentences)
-    return {"embeddings": embeddings.tolist()}
+    return {
+        'data': {
+            'embeddings': embeddings.tolist(),
+            'type': data.type
+        }
+    }
+
+# Hàm Gradio để gọi API FastAPI
+
+
+# async def run_gradio():
+#     demo.launch(share=True)
 
-# Include router
-app.include_router(router)
+async def run_uvicorn():
+    config = uvicorn.Config("app:app", host="0.0.0.0", port=8000, reload=True)
+    server = uvicorn.Server(config)
+    await server.serve()
 
-# Define main function to run the app
-def main():
-    uvicorn.run("api:app", host="0.0.0.0", port=8000, reload=True)
+# async def main():
+#     await asyncio.gather(run_uvicorn(), run_gradio())
 
-# Run the app if this script is the main module
+# Khởi động server
 if __name__ == "__main__":
-    main()
+    asyncio.run(run_uvicorn())