Update app.py

app.py

@@ -1,43 +1,41 @@
 # app.py
-import gradio as gr
+import onnxruntime as ort
 from embeddingonnx import text_to_embedding, query_to_embedding
+import gradio as gr
 
 # ==============================
-# دوال النقاط النهائية باستخدام Gradio
+# دالة سريعة للتحويل (Gradio)
 # ==============================
-
-def embed_endpoint(text: str):
-    """تحويل النص العربي إلى embedding"""
+def embed_text(text: str):
     if not text.strip():
         return "❌ النص فارغ"
     vec = text_to_embedding(text.strip())
-    return vec.tolist()  # إرجاع قائمة من الأرقام
+    return vec.tolist()
 
-def query_endpoint(text: str):
-    """تحويل نص البحث إلى query embedding"""
+def query_text(text: str):
     if not text.strip():
         return "❌ النص فارغ"
     vec = query_to_embedding(text.strip())
     return vec.tolist()
 
 # ==============================
-# واجهة المستخدم Gradio
+# واجهة Gradio بسيطة
 # ==============================
-with gr.Blocks(title="Arabic Text Embedding") as demo:
-    gr.Markdown("## قم بإدخال نص عربي للحصول على متجه embedding الخاص به.")
-    
-    with gr.Tab("Embed Text"):
-        input_text = gr.Textbox(label="النص")
-        output_vec = gr.Textbox(label="Embedding")
-        input_text.submit(embed_endpoint, input_text, output_vec)
-    
-    with gr.Tab("Query Embedding"):
-        query_text = gr.Textbox(label="نص البحث")
-        output_query = gr.Textbox(label="Query Embedding")
-        query_text.submit(query_endpoint, query_text, output_query)
+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)
 
 # ==============================
-# تشغيل Gradio
+# تشغيل محلي
 # ==============================
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)