Update app.py

app.py

@@ -45,7 +45,7 @@ async def extract_code_blocks(markdown_text):
     
     return code_blocks
 
-async def generate_infographic(request: InfographicRequest):
+def generate_infographic(request: InfographicRequest):
     description = request.description
     prompt = PROMPT_TEMPLATE.format(description=description)
     generated_completion = client.chat.completions.create(
@@ -62,8 +62,7 @@ async def generate_infographic(request: InfographicRequest):
         )
     generated_text = generated_completion.choices[0].message.content
     print(generated_text)
-    code_blocks= await extract_code_blocks(generated_text)
-    return code_blocks
+    return generated_text
 
 # Route to serve the HTML template
 @app.get("/", response_class=HTMLResponse)
@@ -73,7 +72,8 @@ async def serve_frontend():
 # Route to handle infographic generation
 @app.post("/generate")
 async def generate_infographic(request: InfographicRequest):
-    code_blocks= await generate_infographic(request)
+    generated_text= generate_infographic(request)
+    code_blocks=await extract_code_blocks(generated_text)
     if code_blocks:
         return JSONResponse(content={"html": code_blocks[0]})
     else: