Refactor proposal generation to use DOCX template

src/app.py

@@ -5,6 +5,8 @@ from enum import Enum
 import pypandoc
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
+from docx import Document
+from docx.shared import Pt
 
 from src.utils import EmailClient
 
@@ -12,13 +14,8 @@ app = FastAPI()
 
 pypandoc.download_pandoc()
 
-TEMPLATE = ""
-with open("src/proposal_template.md", "r") as file:
-    TEMPLATE = file.read()
-
-TEMPLATE = TEMPLATE.replace("`", "")
-VARIABLES = re.findall(r"\{(.*?)\}", TEMPLATE)
-
+template_path = "Template.docx"
+DOC = Document(template_path)
 
 class Email(str, Enum):
     sukhwinder = "sukhwinder@sifars.com"
@@ -45,28 +42,33 @@ class ProposalRequest(BaseModel):
 
 
 async def generate_proposal(data: dict):
-    proposal = TEMPLATE
-    for variable in VARIABLES:
-        proposal = proposal.replace(
-            f"{{{variable}}}",
-            data.get(variable, f"DATA NOT FOUND FOR {variable}. FIXME."),
+
+    for para in DOC.paragraphs:
+        for key, value in data.items():
+            if f"{{{key}}}" in para.text:
+                para.text = para.text.replace(f"{{{key}}}", value)
+                for run in para.runs:
+                    run.font.size = Pt(14)
+                    run.font.name = "Arial"
+
+
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".docx") as temp_file:
+        DOC.save(temp_file.name)
+        docx_file_path = temp_file.name
+
+    attachments = [docx_file_path]
+    recipient_email = data.get("recipient_email", "sukhwinder@sifars.com")
+    subject = f"Proposal for {data.get('project_name')}"
+    body = f"Hello,\n\nPlease find the proposal attached.\n\nBest,\nProposal Bot"
+
+    async with EmailClient() as email_client:
+        await email_client.send_email(
+            subject=subject,
+            body=body,
+            to_emails=[recipient_email],
+            attachments=attachments,
         )
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".md") as temp_file:
-        temp_file.write(proposal.encode())
-        temp_file.flush()
-        docx_file_path = temp_file.name.replace(".md", ".docx")
-        pypandoc.convert_file(temp_file.name, "docx", outputfile=docx_file_path)
-        attachments = [docx_file_path]
-        recipient_email = data.get("recipient_email", "sukhwinder@sifars.com")
-        subject = f"Proposal for {data.get('project_name')}"
-        body = f"Hello,\n\nPlease find the proposal attached.\n\nBest,\nProposal Bot"
-        async with EmailClient() as email_client:
-            await email_client.send_email(
-                subject=subject,
-                body=body,
-                to_emails=[recipient_email],
-                attachments=attachments,
-            )
+
     return {"message": "Proposal sent successfully"}