Update app.py

app.py

@@ -27,7 +27,8 @@ async def generate_code(websocket: WebSocket):
             language = request_data.get("language")
             
             # Build the prompt for ASIC design code generation
-            prompt = f"Generate {language} code for the following ASIC design description:\n\n{description}"
+            prompt = f"Generate {language} code only for the following ASIC design description. Provide the code without any additional explanation or comments.\n\n\n{description}"
+
             messages = [{"role": "user", "content": prompt}]
             
             # Stream content as it is generated
@@ -36,7 +37,7 @@ async def generate_code(websocket: WebSocket):
                 model="Qwen/Qwen2.5-Coder-32B-Instruct",
                 messages=messages,
                 temperature=0.5,
-                max_tokens=1024,
+                max_tokens=4000,
                 top_p=0.7,
                 stream=True
             )
@@ -60,7 +61,7 @@ async def refine_code(websocket: WebSocket):
             language = request_data.get("language")
             
             # Build the refinement prompt
-            prompt = f"Refine the following {language} code based on additional details:\n\n{existing_code}"
+            prompt = f"Refine the following {language} code based on the provided details. Return only the updated code, without explanations or comments.\n\n\n{existing_code}"
             messages = [{"role": "user", "content": prompt}]
             
             # Stream refined code content
@@ -69,7 +70,7 @@ async def refine_code(websocket: WebSocket):
                 model="Qwen/Qwen2.5-Coder-32B-Instruct",
                 messages=messages,
                 temperature=0.5,
-                max_tokens=1024,
+                max_tokens=4000,
                 top_p=0.7,
                 stream=True
             )
@@ -98,7 +99,7 @@ async def optimize_code(websocket: WebSocket):
             model="Qwen/Qwen2.5-Coder-32B-Instruct",
             messages=messages,
             temperature=0.5,
-            max_tokens=512,
+            max_tokens=4000,
             top_p=0.7,
             stream=True
         )