Refactor imports and enhance proposal generation with architecture diagram handling

src/app.py

@@ -1,13 +1,14 @@
+import base64
 import re
 import tempfile
-import base64
-import httpx
 from enum import Enum
+
+import httpx
+import pypandoc
+from docx import Document
+from docx.shared import Inches, Pt
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-from docx import Document
-from docx.shared import Pt, Inches
-import pypandoc
 
 from src.utils import EmailClient
 
@@ -18,11 +19,13 @@ pypandoc.download_pandoc()
 template_path = "src/template.docx"
 DOC = Document(template_path)
 
+
 class Email(str, Enum):
     sukhwinder = "sukhwinder@sifars.com"
     munish = "munish@sifars.com"
     jatin = "jatin@sifars.com"
 
+
 class ProposalRequest(BaseModel):
     project_name: str
     project_overview: str
@@ -41,10 +44,12 @@ class ProposalRequest(BaseModel):
     architecture_diagram: str
     recipient_email: Email
 
+
 async def generate_proposal(data: dict):
 
     mermaid_code = data.get("architecture_diagram", "")
     architecture_diagram_path = None
+    mermaid_code.replace("\n", "")
 
     if mermaid_code.strip():
         graphbytes = mermaid_code.encode("utf8")
@@ -57,13 +62,15 @@ async def generate_proposal(data: dict):
                 response = await client.get(url)
                 response.raise_for_status()
 
-                with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
+                with tempfile.NamedTemporaryFile(
+                    delete=False, suffix=".png"
+                ) as temp_file:
                     temp_file.write(response.content)
                     architecture_diagram_path = temp_file.name
         except httpx.HTTPError as e:
             raise HTTPException(
                 status_code=500,
-                detail=f"Failed to generate architecture diagram: {str(e)}"
+                detail=f"Failed to generate architecture diagram: {str(e)}",
             )
 
     for para in DOC.paragraphs: