Enhance proposal generation by fetching architecture diagrams from Mermaid API and improving image handling

src/app.py

@@ -1,14 +1,13 @@
 import re
 import tempfile
+import base64
+import httpx  # Add this import
 from enum import Enum
-from langchain_core.runnables.graph_mermaid import draw_mermaid_png
-
-import pypandoc
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from docx import Document
-from docx.shared import Pt
-from docx.shared import Inches
+from docx.shared import Pt, Inches  # Ensure Inches is imported
+import pypandoc
 
 from src.utils import EmailClient
 
@@ -24,7 +23,6 @@ class Email(str, Enum):
     munish = "munish@sifars.com"
     jatin = "jatin@sifars.com"
 
-
 class ProposalRequest(BaseModel):
     project_name: str
     project_overview: str
@@ -43,28 +41,47 @@ class ProposalRequest(BaseModel):
     architecture_diagram: str
     recipient_email: Email
 
-
 async def generate_proposal(data: dict):
 
-    # Generate architecture diagram
     mermaid_code = data.get("architecture_diagram", "")
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_image_file:
-        draw_mermaid_png(mermaid_syntax=mermaid_code, output_file_path=temp_image_file.name)
-        architecture_diagram_path = temp_image_file.name
+    architecture_diagram_path = None
+
+    if mermaid_code.strip():
+        graphbytes = mermaid_code.encode("utf8")
+        base64_bytes = base64.urlsafe_b64encode(graphbytes)
+        base64_str = base64_bytes.decode("ascii")
+        url = f"https://mermaid.ink/img/{base64_str}"
+
+        try:
+            async with httpx.AsyncClient() as client:
+                response = await client.get(url)
+                response.raise_for_status()
+                
+                # Save image to 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)}"
+            )
 
     for para in DOC.paragraphs:
         for key, value in data.items():
             if f"{{{key}}}" in para.text:
                 if key == "architecture_diagram":
-                    para.clear()
-                    run = para.add_run()
-                    run.add_picture(architecture_diagram_path)
+                    if architecture_diagram_path:
+                        para.clear()
+                        run = para.add_run()
+                        run.add_picture(architecture_diagram_path, width=Inches(6))
                 else:
                     para.text = para.text.replace(f"{{{key}}}", value)
                     for run in para.runs:
                         run.font.size = Pt(14)
                         run.font.name = "Arial"
 
+    # Save DOCX and send email (unchanged)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".docx") as temp_file:
         DOC.save(temp_file.name)
         docx_file_path = temp_file.name
@@ -72,7 +89,7 @@ async def generate_proposal(data: dict):
     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"
+    body = "Hello,\n\nPlease find the proposal attached.\n\nBest,\nProposal Bot"
 
     async with EmailClient() as email_client:
         await email_client.send_email(