Update pandas_expression_generator_function.py

pandas_expression_generator/src/pandas_expression_generator/pandas_expression_generator_function.py

@@ -54,9 +54,10 @@ async def pandas_expression_generator_function(
     def expression_executor(input_exp: str, df=None) -> tuple:
         # Convert the input string to a list
         commands = ast.literal_eval(input_exp)
-
+        print(input_exp)
         # local_vars = dict([(str.strip, command.split('=', 1)) for command in commands])
         local_vars:dict = dict([list(map(str.strip, command.split('=', 1))) for command in commands])
+        print("THIS")
 
         # Dictionary to simulate local scope
         local_vars.update({'df': df, 'pd': pd, 'np': np})
@@ -107,6 +108,8 @@ async def pandas_expression_generator_function(
                                                 // reset the index of df if requrired
                                                 // convert the grouped dataframe to dictionaries
                                                 // 
+                                                // the last step should always assign output to the variable `output`
+                                                // never use 'print' in the expression
             "output" = sting.  Have the string as list (records) of dictionaries (columns). Ensure to reset index if required
         ]
 
@@ -129,6 +132,7 @@ async def pandas_expression_generator_function(
             structured_text_str = extract_bracket_content(ai_message.content)
             output_message = str(expression_executor(structured_text_str, df))[:800]
             #
+            print("*"*80, output_message)
             prompt_ = f'''You are part of big system. Your task is to respond the user in conversational way.
         
                 This was the query of the end users: