Update src/streamlit_app.py

src/streamlit_app.py

@@ -15,21 +15,28 @@ JSON:"""
 def extract(text: str):
     out = gen(prompt.format(text=text))
     raw = out[0].get("generated_text") or out[0].get("text") or str(out[0])
-    m = re.search(r"\{[\s\S]*\}", raw)
+    m = re.search(r"(\{[\s\S]*\})", raw)
     data = {}
     if m:
-        blob = m.group(0)
+        blob = m.group(0).strip()
         for parser in (json.loads, ast.literal_eval):
             try:
-                data = parser(blob); break
-            except Exception: pass
+                parsed_data = parser(blob)
+                if isinstance(parsed_data, list) and parsed_data:
+                    data = parsed_data[0]
+                elif isinstance(parsed_data, dict):
+                    data = parsed_data
+                break
+            except Exception:
+                continue
+
     if not isinstance(data, dict):
         return {
             "SKILL": ["(Error: Invalid/Corrupted Model Output)"],
             "KNOWLEDGE": [],
             "DEBUG_RAW_OUTPUT": raw
         }
-    
+
     return {
         "SKILL": data.get("SKILL", []),
         "KNOWLEDGE": data.get("KNOWLEDGE", [])
@@ -37,5 +44,6 @@ def extract(text: str):
 
 st.title("Skill/Knowledge Extractor")
 text = st.text_area("Paste text")
+
 if st.button("Extract") and text.strip():
     st.json(extract(text))