Update app.py

app.py

@@ -2,11 +2,11 @@ import gradio as gr
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
-from embeddingonnx import text_to_embedding, query_to_embedding
+from embeddingonnx import text_to_embedding
 
-app = FastAPI(title="Arabic Text Embedding API")
+# -------- FastAPI --------
+app = FastAPI(title="Arabic Embedding API")
 
-# ------------- CORS -------------
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -15,68 +15,29 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# ----------- Models ----------
 class TextRequest(BaseModel):
     text: str
 
-
 @app.get("/")
 def root():
-    return {"message": "Arabic Embedding API is running on HF Spaces ✔"}
-
-
-@app.get("/health")
-def health():
-    return {"status": "ok"}
-
+    return {"message": "Arabic Embedding API is running ✔"}
 
 @app.post("/embed")
 def embed_post(req: TextRequest):
     text = req.text.strip()
     if not text:
         raise HTTPException(400, "النص فارغ")
-
     vec = text_to_embedding(text, normalize=True)
     return {"embedding": vec.tolist()}
 
-
-@app.post("/query")
-def query_post(req: TextRequest):
-    text = req.text.strip()
-    if not text:
-        raise HTTPException(400, "النص فارغ")
-
-    vec = query_to_embedding(text, normalize=True)
-    return {"query_embedding": vec.tolist()}
-
-
-# ------ GET Versions ------
-@app.get("/embed")
-def embed_get(text: str):
-    text = text.strip()
-    if not text:
-        raise HTTPException(400, "النص فارغ")
-    vec = text_to_embedding(text, normalize=True)
-    return {"embedding": vec.tolist()}
-
-
-@app.get("/query")
-def query_get(text: str):
-    text = text.strip()
-    if not text:
-        raise HTTPException(400, "النص فارغ")
-    vec = query_to_embedding(text, normalize=True)
-    return {"query_embedding": vec.tolist()}
-
-
-# ========== Gradio UI ==========
-def ui_fn(text):
+# -------- Gradio UI --------
+def gr_ui(text):
     if not text.strip():
         return "❌ النص فارغ"
     vec = text_to_embedding(text, normalize=True)
     return str(vec.tolist())
 
+demo = gr.Interface(fn=gr_ui, inputs="text", outputs="text", title="Arabic Embedding API")
 
-demo = gr.Interface(fn=ui_fn, inputs="text", outputs="text", title="Arabic Embedding API")
-
+# ربط Gradio مع FastAPI
 gr.mount_gradio_app(app, demo, path="/gradio")