Update main.py

main.py

@@ -76,7 +76,19 @@ def bot():
     data = eval(str(json_table))
     df = pd.DataFrame(data)
     print(list(df))
-    pandas_agent = SmartDataframe(df,config={"llm":llm, "response_parser":FlaskResponse})
+    pandas_agent = SmartDataframe(df,config={"llm":llm, "response_parser":FlaskResponse}, "custom_whitelisted_dependencies": [
+        "os",
+        "io",
+        "sys",
+        "chr",
+        "glob",
+        "b64decoder",
+        "collections",
+        "geopy",
+        "geopandas",
+        "wordcloud",
+        "builtins"
+    ], "security":"none")
     answer = pandas_agent.chat(user_question)
     return jsonify(answer)
     return answer