Update app.py

app.py

@@ -31,7 +31,6 @@ def generate_solution(api_key, problem_statement):
         return f"Error: {str(e)}"
 
 def format_response(response):
-    # Split code and explanation
     formatted = []
     in_code = False
     code_buffer = []
@@ -39,10 +38,12 @@ def format_response(response):
     
     for line in response.split('\n'):
         if line.strip().startswith('```'):
-            in_code = not in_code
-            if not in_code:
-                formatted.append(f"```python\n{'\n'.join(code_buffer[1:])}\n```")
+            if in_code and code_buffer:
+                # Fix: Separate string concatenation
+                code_content = '\n'.join(code_buffer[1:])  # Remove first line (language marker)
+                formatted.append(f"```python\n{code_content}\n```")
                 code_buffer = []
+            in_code = not in_code
             continue
             
         if in_code: