UPDATE: Analyzer

app.py

@@ -165,14 +165,14 @@ async def getYTTranscript(url: str):
 
 @app.post("/analyzeData")
 async def analyzeAndAnswer(query: str, file: UploadFile = File(...)):
- extension = file.name.split(".")[-1]
+ extension = file.filename.split(".")[-1]
  if extension in ["xls", "xlsx", "xlsm", "xlsb"]:
- df = pd.read_excel(io.BytesIO(file.read()))
+ df = pd.read_excel(io.BytesIO(await file.read()))
  return {
  "output": analyzeData(query = query, dataframe = df)
  }
  elif extension == "csv":
- df = pd.read_csv(io.BytesIO(file.read()))
+ df = pd.read_csv(io.BytesIO(await file.read()))
  return {
  "output": analyzeData(query = query, dataframe = df)
  }

functions.py

@@ -298,10 +298,9 @@ def getTextFromImagePDF(pdfBytes):
  text = "\n\n\n".join(["\n".join([text[1] for text in reader.readtext(image, paragraph=True)]) for image in allImages])
  return text
 
-
-def getTranscript(urls: dict[str, str]):
+def getTranscript(urls: str):
  loader = YoutubeLoader.from_youtube_url(
- url, add_video_info=False
+ urls.split(","), add_video_info = False
  )
  try:
  doc = " ".join([x.page_content for x in loader.load()])
@@ -311,7 +310,7 @@ def getTranscript(urls: dict[str, str]):
 
 
 def analyzeData(query, dataframe):
- llm = ChatGroq("gemma2-9b-it")
+ llm = ChatGroq(name = "gemma2-9b-it")
  df = SmartDataframe(dataframe, config = {"llm": llm, "verbose": False})
  response = df.chat(query)
  return response