Update app.py

app.py

@@ -23,7 +23,7 @@ class InfographicRequest(BaseModel):
 PROMPT_TEMPLATE = os.getenv("PROMPT_TEMPLATE")
 
 
-def extract_code_blocks(markdown_text):
+async def extract_code_blocks(markdown_text):
     """
     Extracts code blocks from the given Markdown text.
 
@@ -67,8 +67,9 @@ async def generate_infographic(request: InfographicRequest):
         generated_text = ""
         for chunk in stream:
             generated_text += chunk.choices[0].delta.content
-        
-        code_blocks=extract_code_blocks(generated_text)
+
+        print(generated_text)
+        code_blocks= await extract_code_blocks(generated_text)
         if code_blocks:
             return JSONResponse(content={"html": code_blocks[0]})
         else: