UPDATE: Analyzer

app.py

@@ -166,17 +166,19 @@ async def getYTTranscript(url: str):
 @app.post("/analyzeData")
 async def analyzeAndAnswer(query: str, file: UploadFile = File(...)):
     extension = file.filename.split(".")[-1]
-    if extension in ["xls", "xlsx", "xlsm", "xlsb"]:
-        df = pd.read_excel(io.BytesIO(await file.read()))
+    try:
+        if extension in ["xls", "xlsx", "xlsm", "xlsb"]:
+            df = pd.read_excel(io.BytesIO(await file.read()))
+            response = analyzeData(query = query, dataframe = df)
+        elif extension == "csv":
+            df = pd.read_csv(io.BytesIO(await file.read()))
+            response = analyzeData(query = query, dataframe = df)
+        else:
+            response = "INVALID FILE TYPE"
         return {
-            "output": analyzeData(query = query, dataframe = df)
-            }
-    elif extension == "csv":
-        df = pd.read_csv(io.BytesIO(await file.read()))
-        return {
-            "output": analyzeData(query = query, dataframe = df)
-            }
-    else:
+            "output": response
+        }
+    except:
         return {
-            "output": "INVALID FILE TYPE"
+            "output": "UNABLE TO ANSWER QUERY"
         }