Update app.py

app.py

@@ -29,9 +29,6 @@ index = pc.Index(PINECONE_INDEX)
 embedder = SentenceTransformer("all-mpnet-base-v2")
 api = HfApi()
 
-# ----------------- EMBEDDER -----------------
-embedder = SentenceTransformer("all-mpnet-base-v2")
-
 # ----------------- HELPERS -----------------
 def load_reference_text(uploaded_file) -> str:
     if not uploaded_file:
@@ -64,14 +61,14 @@ def load_templates_json() -> List[Dict]:
     except Exception:
         return []
 
-def parse_json_safe(raw_text: str, fallback: str) -> List[str]:
-    try:
-        return json.loads(raw_text)
-    except:
-        matches = re.findall(r'"([^"]+)"', raw_text)
-        if matches:
-            return matches
-        return [fallback[:512]]
+# def parse_json_safe(raw_text: str, fallback: str) -> List[str]:
+#     try:
+#         return json.loads(raw_text)
+#     except:
+#         matches = re.findall(r'"([^"]+)"', raw_text)
+#         if matches:
+#             return matches
+#         return [fallback[:512]]
 
 def save_template_to_hf(name: str, analysis: str) -> Tuple[bool, str]:
     """Save new template into HF dataset repo (templates.json)."""
@@ -235,7 +232,7 @@ def generate_legal_draft(case_text, uploaded_file, template_name, new_template_n
     yield gr.update(value="🔍 Searching in Knowledge Base..."), None
 
     queries = build_queries_with_llm(case_text)
-    context_text, citations = pinecone_search(queries, top_k=10)
+    context_text, citations = pinecone_search(queries, top_k=15)
 
     # Handle template
     template_analysis = ""