update app.py

app.py

@@ -1,12 +1,17 @@
+# app.py
 import gradio as gr
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Query
 from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
-from embeddingonnx import text_to_embedding
+from embeddingonnx import text_to_embedding, query_to_embedding
+import numpy as np
 
-# -------- FastAPI --------
-app = FastAPI(title="Arabic Embedding API")
+# ==============================
+# إنشاء تطبيق FastAPI
+# ==============================
+app = FastAPI(title="Arabic Text Embedding API")
 
+# ------------- CORS -------------
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -15,12 +20,22 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# ==============================
+# نموذج البيانات الوارد
+# ==============================
 class TextRequest(BaseModel):
     text: str
 
+# ==============================
+# نقاط النهاية REST API
+# ==============================
 @app.get("/")
 def root():
-    return {"message": "Arabic Embedding API is running ✔"}
+    return {"message": "✅ Arabic Text Embedding API is running on HF Spaces"}
+
+@app.get("/health")
+def health():
+    return {"status": "ok"}
 
 @app.post("/embed")
 def embed_post(req: TextRequest):
@@ -30,14 +45,53 @@ def embed_post(req: TextRequest):
     vec = text_to_embedding(text, normalize=True)
     return {"embedding": vec.tolist()}
 
-# -------- Gradio UI --------
-def gr_ui(text):
+@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()}
+
+@app.get("/embed")
+def embed_get(text: str = Query(..., description="النص المراد تحويله إلى embedding")):
+    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 = Query(..., description="النص المراد تحويله إلى query embedding")):
+    text = text.strip()
+    if not text:
+        raise HTTPException(400, "النص فارغ")
+    vec = query_to_embedding(text, normalize=True)
+    return {"query_embedding": vec.tolist()}
+
+# ==============================
+# واجهة Gradio
+# ==============================
+def gradio_embed(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=gradio_embed,
+    inputs="text",
+    outputs="text",
+    title="Arabic Text Embedding",
+    description="قم بإدخال نص عربي للحصول على متجه embedding الخاص به."
+)
 
-# ربط Gradio مع FastAPI
+# ربط Gradio مع FastAPI على المسار /gradio
 gr.mount_gradio_app(app, demo, path="/gradio")
+
+# ==============================
+# تشغيل السيرفر محلياً (اختياري)
+# ==============================
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)