Update app.py

app.py

@@ -45,7 +45,16 @@ async def extract_code_blocks(markdown_text):
     
     return code_blocks
 
-def generate_infographic(request: InfographicRequest):
+
+
+# Route to serve the HTML template
+@app.get("/", response_class=HTMLResponse)
+async def serve_frontend():
+    return HTMLResponse(open("static/infographic_gen.html").read())
+
+# Route to handle infographic generation
+@app.post("/generate")
+async def generate_infographic(request: InfographicRequest):
     description = request.description
     prompt = PROMPT_TEMPLATE.format(description=description)
     generated_completion = client.chat.completions.create(
@@ -61,18 +70,6 @@ def generate_infographic(request: InfographicRequest):
             stop=None
         )
     generated_text = generated_completion.choices[0].message.content
-    print(generated_text)
-    return generated_text
-
-# Route to serve the HTML template
-@app.get("/", response_class=HTMLResponse)
-async def serve_frontend():
-    return HTMLResponse(open("static/infographic_gen.html").read())
-
-# Route to handle infographic generation
-@app.post("/generate")
-async def generate_infographic(request: InfographicRequest):
-    generated_text= generate_infographic(request)
     code_blocks=await extract_code_blocks(generated_text)
     if code_blocks:
         return JSONResponse(content={"html": code_blocks[0]})