Update app.py

app.py

@@ -103,24 +103,8 @@ def pinecone_search(queries: List[str], top_k: int = 10, max_chars: int = 10000)
                 break
     return "\n".join(context_parts), citations
 
-# def markdown_to_docx(md_text: str) -> str:
-#     """
-#     Convert Markdown text into a temporary .docx file and return its path.
-#     """
-#     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".docx")
-#     doc = Document()
-#     for line in md_text.split("\n"):
-#         if line.startswith("## "):
-#             doc.add_heading(line[3:], level=2)
-#         elif line.startswith("# "):
-#             doc.add_heading(line[2:], level=1)
-#         else:
-#             doc.add_paragraph(line)
-#     doc.save(temp_file.name)
-#     return temp_file.name
-    
-def markdown_to_docx(md_text: str) -> BytesIO:
-    """Convert Markdown text into Word DOCX with proper formatting."""
+def markdown_to_docx(md_text: str) -> str:
+    """Convert Markdown text into a Word DOCX and return a file path."""
     html = markdown.markdown(md_text)
     soup = BeautifulSoup(html, "html.parser")
 
@@ -137,10 +121,33 @@ def markdown_to_docx(md_text: str) -> BytesIO:
         elif el.name == "li":
             doc.add_paragraph(f"• {el.get_text()}")
 
-    buf = BytesIO()
-    doc.save(buf)
-    buf.seek(0)
-    return buf
+    # Save into a temporary file
+    tmp_path = os.path.join(tempfile.gettempdir(), "draft.docx")
+    doc.save(tmp_path)
+    return tmp_path
+    
+# def markdown_to_docx(md_text: str) -> BytesIO:
+#     """Convert Markdown text into Word DOCX with proper formatting."""
+#     html = markdown.markdown(md_text)
+#     soup = BeautifulSoup(html, "html.parser")
+
+#     doc = Document()
+#     for el in soup.descendants:
+#         if el.name == "h1":
+#             doc.add_heading(el.get_text(), level=1)
+#         elif el.name == "h2":
+#             doc.add_heading(el.get_text(), level=2)
+#         elif el.name == "h3":
+#             doc.add_heading(el.get_text(), level=3)
+#         elif el.name == "p":
+#             doc.add_paragraph(el.get_text())
+#         elif el.name == "li":
+#             doc.add_paragraph(f"• {el.get_text()}")
+
+#     buf = BytesIO()
+#     doc.save(buf)
+#     buf.seek(0)
+#     return buf
 
 
 # ----------------- MAIN FUNCTION -----------------
@@ -212,6 +219,7 @@ def generate_legal_draft(case_text, uploaded_file=None, add_citations=True):
     return gr.update(value=draft_md), markdown_to_docx(draft_md)
 
 
+
 # ----------------- GRADIO INTERFACE -----------------
 with gr.Blocks() as demo:
     gr.Markdown("## ⚖️ AI Legal Draft Generator\nUpload a DOCX/PDF/TXT template and enter case details. Get a court-ready draft.")