Update app.py

app.py

@@ -73,9 +73,8 @@ async def serve_frontend():
 async def generate_infographic(request: InfographicRequest):
     description =await generate_infographic_details(request)
     prompt = PROMPT_TEMPLATE.format(description=description)
-    
-    try:
-        messages = [{"role": "user", "content": prompt}]
+
+    messages = [{"role": "user", "content": prompt}]
         stream = client.chat.completions.create(
             model="Qwen/Qwen2.5-Coder-32B-Instruct",
             messages=messages,
@@ -86,16 +85,39 @@ async def generate_infographic(request: InfographicRequest):
         )
 
         
-        generated_text = ""
-        for chunk in stream:
-            generated_text += chunk.choices[0].delta.content
-
-        print(generated_text)
-        code_blocks= await extract_code_blocks(generated_text)
-        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)
+    generated_text = ""
+    for chunk in stream:
+        generated_text += chunk.choices[0].delta.content
+
+    print(generated_text)
+    code_blocks= await extract_code_blocks(generated_text)
+    if code_blocks:
+        return JSONResponse(content={"html": code_blocks[0]})
+    else:
+        return JSONResponse(content={"error": "No generation"},status_code=500)
+    
+    # try:
+    #     messages = [{"role": "user", "content": prompt}]
+    #     stream = client.chat.completions.create(
+    #         model="Qwen/Qwen2.5-Coder-32B-Instruct",
+    #         messages=messages,
+    #         temperature=0.4,
+    #         max_tokens=6000,
+    #         top_p=0.7,
+    #         stream=True,
+    #     )
+
+        
+    #     generated_text = ""
+    #     for chunk in stream:
+    #         generated_text += chunk.choices[0].delta.content
+
+    #     print(generated_text)
+    #     code_blocks= await extract_code_blocks(generated_text)
+    #     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)