Update app.py

app.py

@@ -74,8 +74,13 @@ 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)
+        if code_blocks:
+            return JSONResponse(content={"html": code_blocks[0]})
+        else:
+            return JSONResponse(content={"error": "No generation"},status_code=500)
 
-    try:
+    # try:
         # messages = [{"role": "user", "content": prompt}]
         # stream = client.chat.completions.create(
         #     model="Qwen/Qwen2.5-Coder-32B-Instruct",
@@ -93,11 +98,11 @@ async def generate_infographic(request: InfographicRequest):
 
         # print(generated_text)
         #code_blocks= await extract_code_blocks(generated_text)
-        code_blocks= await generate_infographic(description)
-        if code_blocks:
-            return JSONResponse(content={"html": code_blocks[0]})
-        else:
-            return JSONResponse(content={"error": "No generation"},status_code=500)
-
-    except Exception as e:
-        return JSONResponse(content={"error": str(e)}, status_code=500)
+        # code_blocks= await generate_infographic(description)
+        # if code_blocks:
+        #     return JSONResponse(content={"html": code_blocks[0]})
+        # else:
+        #     return JSONResponse(content={"error": "No generation"},status_code=500)
+
+    # except Exception as e:
+    #     return JSONResponse(content={"error": str(e)}, status_code=500)