Update app.py

app.py

@@ -45,12 +45,12 @@ async def extract_code_blocks(markdown_text):
     
     return code_blocks
 
-async def generate_infographic(infoRequest):
+async def generate_infographic(prompt):
     generated_completion = client.chat.completions.create(
             model="llama-3.1-70b-versatile",
             messages=[
                 {"role": "system", "content": SYSTEM_INSTRUCT},
-                {"role": "user", "content": infoRequest}
+                {"role": "user", "content": prompt}
             ],
             temperature=0.5,
             max_tokens=5000,
@@ -73,7 +73,7 @@ async def serve_frontend():
 async def generate_infographic(request: InfographicRequest):
     description = request.description
     prompt = PROMPT_TEMPLATE.format(description=description)
-    code_blocks= await generate_infographic(description)
+    code_blocks= await generate_infographic(prompt)
     if code_blocks:
         return JSONResponse(content={"html": code_blocks[0]})
     else: