Update stories.py

stories.py

@@ -85,11 +85,6 @@ class FlaskResponse(ResponseParser):
         # For non-image responses, simply return the value as a string.
         return str(result["value"])
 
-
-
-
-
-
 # Pandasai gemini
 llm1 = ChatGoogleGenerativeAI(
     model="gemini-2.0-flash-thinking-exp",
@@ -135,6 +130,30 @@ def get_pdf_text(pdf_file):
         text = text[:MAX_CHARACTERS]
     return text
 
+def get_df(uploaded_file, ext):
+    """
+    Reads an uploaded file into a pandas DataFrame if the extension is csv, xlsx, or xls.
+
+    Args:
+        uploaded_file: The uploaded file object.
+        ext (str): The extension of the uploaded file.
+
+    Returns:
+        pandas.DataFrame: The DataFrame if the file is successfully read, otherwise None.
+    """
+    if ext in ["csv", "xlsx", "xls"]:
+        try:
+            if ext == "csv":
+                df = pd.read_csv(uploaded_file)
+            else:
+                df = pd.read_excel(uploaded_file)
+            return df
+        except Exception as e:
+            print(f"Error reading file: {e}")
+            return None
+    else:
+        print(f"Unsupported file extension: {ext}. Please upload a csv, xlsx, or xls file.")
+        return None
 
 # -----------------------
 # Audio Transcription