Otherwa (#15)

* /updates + column qual -> quan, graph + / * /file+ dir setup 30%/ * /test_data/ * /changes/ * /line/ * / + dataframe editor file.py + charts file.py + components folder + moulde folder / * /minor fixes/ * /df fixes/ * /fixes/ * /+req/ * /req/ * /chnages/ --------- Co-authored-by: Otherwa <[email protected]>
Otherwa · Nov 26, 2023 · 2ac1699 · 2ac1699
1 parent 46b94ef
commit 2ac1699
Show file tree

Hide file tree

Showing 6 changed files with 19 additions and 5 deletions.
diff --git a/.gitignore b/.gitignore
@@ -162,4 +162,9 @@ cython_debug/
 # ? SPACE FILES
 .env
 Spacefile
-.space
+<<<<<<< HEAD
+.space
+cache
+=======
+.space
+>>>>>>> 46b94ef52bcb30a222ea766dbdf1197d14331d44
diff --git a/.streamlit/config.toml b/.streamlit/config.toml
@@ -1,4 +1,4 @@
 [theme]
 base="light"
-primaryColor="#e02e2e"
-textColor="#2e2f37"
+primaryColor="#d42fbc"
+textColor="#000000"
diff --git a/cache/cache_db.db b/cache/cache_db.db
diff --git a/data/charts/9e2c317e-480b-4e3d-b163-bef0c7a035b3.png b/data/charts/9e2c317e-480b-4e3d-b163-bef0c7a035b3.png
diff --git a/requirements.txt b/requirements.txt
@@ -8,4 +8,4 @@ streamlit-option-menu
 streamlit-pandas-profiling
 plotly
 pymongo
-tk
+tk
diff --git a/src/modules/file/query.py b/src/modules/file/query.py
@@ -35,7 +35,16 @@ def format_other(self, result):
 def query_csv_transformers(df, user_question, api):
     llm = OpenAI(api_token=openai_api_key)
     if api:
-        pandas_ai = SmartDataframe(df, config={"llm": llm})
+        pandas_ai = SmartDataframe(
+            df,
+            config={
+                "llm": llm,
+                "save_charts": True,
+                "save_charts_path": r"./data/charts",
+            },
+        )
         response = pandas_ai.chat(user_question)
         print(response)
+        if response == None:
+            return {"answer": None}
         return {"answer": response}