Update app.py

app.py

@@ -1,16 +1,21 @@
 import gradio as gr
 from sentence_transformers import SentenceTransformer
+import json
 
 embedding_model = SentenceTransformer("all-MiniLM-L6-v2")
 
-def get_embedding(texts_input: str):
-    import json
-    try:
-        texts = json.loads(texts_input)
-        if not isinstance(texts, list):
-            texts = [texts_input]
-    except:
-        texts = [t.strip() for t in texts_input.split("\n") if t.strip()]
+def get_embedding(texts_input):
+    # Parse input nếu là JSON list
+    if isinstance(texts_input, str):
+        try:
+            texts = json.loads(texts_input)
+            if not isinstance(texts, list):
+                texts = [texts_input]
+        except:
+            # fallback: mỗi dòng 1 chunk
+            texts = [line.strip() for line in texts_input.split("\n") if line.strip()]
+    else:
+        texts = texts_input  # nếu client gửi list thật sự
     
     embeddings = embedding_model.encode(
         texts, convert_to_numpy=True, normalize_embeddings=True
@@ -19,8 +24,8 @@ def get_embedding(texts_input: str):
 
 demo = gr.Interface(
     fn=get_embedding,
-    inputs=gr.Textbox(lines=10, placeholder="JSON list or each line one text"),
-    outputs=gr.JSON(label="Embeddings"),
+    inputs=gr.JSON(label="Input chunks"),  # <- quan trọng: JSON input
+    outputs=gr.JSON(label="Embeddings")
 )
 
 demo.launch()