Update app.py

app.py

@@ -7,6 +7,7 @@ from docx import Document
 from sentence_transformers import SentenceTransformer
 from pinecone import Pinecone
 import openai
+import tempfile
 import json
 import re
 
@@ -98,7 +99,11 @@ 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) -> BytesIO:
+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("## "):
@@ -107,10 +112,8 @@ def markdown_to_docx(md_text: str) -> BytesIO:
             doc.add_heading(line[2:], level=1)
         else:
             doc.add_paragraph(line)
-    buf = BytesIO()
-    doc.save(buf)
-    buf.seek(0)
-    return buf
+    doc.save(temp_file.name)
+    return temp_file.name
 
 # ----------------- MAIN FUNCTION -----------------
 def generate_legal_draft(case_text, uploaded_file=None, add_citations=True):
@@ -177,10 +180,11 @@ Instructions
         for i, c in enumerate(citations, 1):
             draft_md += f"{i}. {c['source']} (score: {c['score']:.3f})\n"
 
-    return draft_md, markdown_to_docx(draft_md)
+    docx_path = markdown_to_docx(draft_md)
+    return draft_md, docx_path
 
 # ----------------- GRADIO INTERFACE -----------------
-with gr.Blocks() as iface:
+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.")
 
     case_text = gr.Textbox(label="Case Details", lines=10, placeholder="Enter client and case info...")
@@ -188,7 +192,7 @@ with gr.Blocks() as iface:
     add_citations = gr.Checkbox(label="Append citations", value=True)
 
     draft_output = gr.Markdown(label="Draft Output")
-    download_btn = gr.DownloadButton(label="⬇️ Download Word", file_name="legal_draft.docx")
+    download_btn = gr.DownloadButton(label="⬇️ Download Word")
 
     btn = gr.Button("Generate Draft")
     btn.click(
@@ -197,5 +201,6 @@ with gr.Blocks() as iface:
         outputs=[draft_output, download_btn],
     )
 
+# ----------------- RUN -----------------
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()