Update app.py

app.py

@@ -8,6 +8,7 @@ import mistune
 import pytz
 import math
 import requests
+import pandas as pd
 
 from datetime import datetime
 from openai import ChatCompletion
@@ -48,6 +49,12 @@ def create_file(filename, prompt, response):
     elif filename.endswith(".md"):
         with open(filename, 'w') as file:
             file.write(f"# Prompt:\n{prompt}\n# Response:\n{response}")
+    elif filename.endswith(".csv"):
+        response_df = pd.DataFrame({"Prompt": [prompt], "Response": [response]})
+        response_df.to_csv(filename, index=False)
+    elif filename.endswith(".xlsx"):
+        response_df = pd.DataFrame({"Prompt": [prompt], "Response": [response]})
+        response_df.to_excel(filename, index=False)
     
 # Updated to auto process transcript to chatgpt in AI pipeline from Whisper to ChatGPT
 def transcribe_audio(openai_key, file_path, model):
@@ -141,6 +148,12 @@ def read_file_content(file,max_length):
         return content
     elif file.type == "text/plain":
         return file.getvalue().decode()
+    elif file.type == "text/csv":
+        df = pd.read_csv(file)
+        return df.to_string(index=False)
+    elif file.type == "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet":
+        df = pd.read_excel(file)
+        return df.to_string(index=False)
     else:
         return ""