Update app.py

app.py

@@ -80,7 +80,7 @@ def generate_code(data: RequestData):
     if match:
         clean_output = match.group(0).strip()
     else:
-        fallback = generated_text.split("return row;")[0] + "return row;"
+        fallback = generated_text.split("return row;")[0] + "return row;\n}"
         clean_output = fallback.strip()
 
     return PlainTextResponse(clean_output)