Update app.py

app.py

@@ -91,13 +91,26 @@ def generate_code(query, language, model, client, complexity="Medium"):
             model=model,
             messages=[{
                 "role": "user",
-                "content": f"{complexity_levels[complexity]} {query} in {language}. Include comments."
+                "content": f"""
+                {complexity_levels[complexity]} {query} in {language}. 
+                Include comments in the code itself.
+                IMPORTANT: Return ONLY the raw executable code with comments, 
+                without any additional explanation before or after the code block.
+                """
             }],
             temperature=0.7 if complexity == "Advanced" else 0.5,
             max_tokens=4096,
             top_p=0.95
         )
-        return completion.choices[0].message.content
+        # Extract just the code block if it's wrapped in markdown
+        raw_content = completion.choices[0].message.content
+        if '```' in raw_content:
+            # Extract content between the first ``` and last ```
+            code = raw_content.split('```')[1]
+            if code.startswith(language.lower()):
+                code = code[len(language.lower()):]
+            return code.strip()
+        return raw_content.strip()
     except Exception as e:
         st.error(f"Error generating code: {str(e)}")
         return None