rairo commited on
Commit
2e7b4ff
·
verified ·
1 Parent(s): 80dd602

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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=gemini_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
  )