Update app.py

app.py

@@ -45,7 +45,9 @@ async def extract_code_blocks(markdown_text):
     
     return code_blocks
 
-async def generate_infographic(prompt):
+async def generate_infographic(request: InfographicRequest):
+    description = request.description
+    prompt = PROMPT_TEMPLATE.format(description=description)
     generated_completion = client.chat.completions.create(
             model="llama-3.1-70b-versatile",
             messages=[
@@ -71,9 +73,7 @@ async def serve_frontend():
 # Route to handle infographic generation
 @app.post("/generate")
 async def generate_infographic(request: InfographicRequest):
-    description = request.description
-    prompt = PROMPT_TEMPLATE.format(description=description)
-    code_blocks= await generate_infographic(prompt)
+    code_blocks= await generate_infographic(request)
     if code_blocks:
         return JSONResponse(content={"html": code_blocks[0]})
     else: