Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,6 +10,7 @@ import os
|
|
| 10 |
import pandas as pd
|
| 11 |
from pandasai import SmartDataframe, SmartDatalake
|
| 12 |
from pandasai.responses.response_parser import ResponseParser
|
|
|
|
| 13 |
|
| 14 |
class StreamLitResponse(ResponseParser):
|
| 15 |
def __init__(self,context) -> None:
|
|
@@ -34,7 +35,7 @@ if not GOOGLE_API_KEY:
|
|
| 34 |
|
| 35 |
|
| 36 |
def generateResponse(prompt, dfs):
|
| 37 |
-
llm = GoogleGemini(api_key=
|
| 38 |
pandas_agent = SmartDatalake(dfs,config={"llm":llm, "response_parser":StreamLitResponse})
|
| 39 |
answer = pandas_agent.chat(prompt)
|
| 40 |
return answer
|
|
@@ -111,6 +112,7 @@ def main():
|
|
| 111 |
|
| 112 |
with st.sidebar:
|
| 113 |
st.subheader("Your files")
|
|
|
|
| 114 |
uploaded_files = st.file_uploader(
|
| 115 |
"Upload PDFs, CSVs, or Excel files (up to 3)", accept_multiple_files=True
|
| 116 |
)
|
|
|
|
| 10 |
import pandas as pd
|
| 11 |
from pandasai import SmartDataframe, SmartDatalake
|
| 12 |
from pandasai.responses.response_parser import ResponseParser
|
| 13 |
+
from pandasai.llm import GoogleGemini
|
| 14 |
|
| 15 |
class StreamLitResponse(ResponseParser):
|
| 16 |
def __init__(self,context) -> None:
|
|
|
|
| 35 |
|
| 36 |
|
| 37 |
def generateResponse(prompt, dfs):
|
| 38 |
+
llm = GoogleGemini(api_key=GOOGLE_API_KEY)
|
| 39 |
pandas_agent = SmartDatalake(dfs,config={"llm":llm, "response_parser":StreamLitResponse})
|
| 40 |
answer = pandas_agent.chat(prompt)
|
| 41 |
return answer
|
|
|
|
| 112 |
|
| 113 |
with st.sidebar:
|
| 114 |
st.subheader("Your files")
|
| 115 |
+
st.sidebar.image("logoqb.jpeg", use_column_width=True)
|
| 116 |
uploaded_files = st.file_uploader(
|
| 117 |
"Upload PDFs, CSVs, or Excel files (up to 3)", accept_multiple_files=True
|
| 118 |
)
|