Format results as table

models/space_er.py

@@ -30,7 +30,7 @@ except Exception as e:
 # Entity labels to extract
 LABELS = ["MEDICATION", "DOSAGE", "DURATION"]
 
-def extract_entities(text: str) -> str:
+def extract_entities(text: str):
     """
     Extract named entities from Arabic text using GLiNER.
 
@@ -38,7 +38,7 @@ def extract_entities(text: str) -> str:
         text (str): Input Arabic text.
 
     Returns:
-        str: Formatted entity extraction result or error message.
+        list[dict] | str: List of extracted entities (as table rows) or error message.
     """
     if not text.strip():
         return "يرجى إدخال نص للتحليل / Please enter text to analyze"
@@ -48,10 +48,8 @@ def extract_entities(text: str) -> str:
         if not entities:
             return "لم يتم العثور على أي كيانات / No entities found"
 
-        result = "الكيانات المستخرجة:\n\n"
-        for ent in entities:
-            result += f"{ent['text']} => {ent['label']}\n"
-        return result
+        # Return as list of dicts for Gradio Dataframe
+        return [{"النص": ent["text"], "التصنيف": ent["label"]} for ent in entities]
 
     except Exception as e:
         return f"خطأ في التحليل / Analysis Error: {str(e)}"