# app.py
import onnxruntime as ort
from embeddingonnx import text_to_embedding, query_to_embedding
import gradio as gr

# ==============================
# دالة سريعة للتحويل (Gradio)
# ==============================
def embed_text(text: str):
    if not text.strip():
        return "❌ النص فارغ"
    vec = text_to_embedding(text.strip())
    return vec.tolist()

def query_text(text: str):
    if not text.strip():
        return "❌ النص فارغ"
    vec = query_to_embedding(text.strip())
    return vec.tolist()

# ==============================
# واجهة Gradio بسيطة
# ==============================
with gr.Blocks() as demo:
    gr.Markdown("## Arabic Text Embedding API (Local)")

    with gr.Row():
        text_input = gr.Textbox(label="النص", placeholder="أدخل نص عربي...")
        text_output = gr.Textbox(label="Embedding")
        text_input.submit(embed_text, text_input, text_output)

    with gr.Row():
        query_input = gr.Textbox(label="Query", placeholder="أدخل نص البحث...")
        query_output = gr.Textbox(label="Query Embedding")
        query_input.submit(query_text, query_input, query_output)

# ==============================
# تشغيل محلي
# ==============================
# ==============================
# تشغيل محلي مع رابط 


if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7860)