Update csv_agent.py

csv_agent.py

@@ -7,16 +7,17 @@ client = InferenceClient(
 )
 
 def generate_code(prompt):
-    full_prompt = f"""You are a Python Pandas expert.
-Write Python Pandas code to perform the following operation on a dataframe named df:
-{prompt}
-Only output valid Python code, no explanation, no extra text."""
+    full_prompt = f"""You are a professional Python Pandas developer.
+Write a complete, executable, valid Python Pandas script to perform the following task on a DataFrame named df:
+Task: {prompt}
+Make sure the code can be executed without any syntax errors.
+Output only valid Python code. No explanation, no comments."""
 
     response = client.text_generation(
         prompt=full_prompt,
-        max_new_tokens=200,
+        max_new_tokens=200,     # Updated token limit
         temperature=0.2,
         top_p=0.9,
         repetition_penalty=1.05
     )
-    return response
+    return response