Update app.py

app.py

@@ -19,7 +19,7 @@ from audio_recorder_streamlit import audio_recorder
 openai.api_key = os.getenv('OPENAI_KEY')
 st.set_page_config(page_title="GPT Streamlit Document Reasoner",layout="wide")
 
-menu = ["htm", "txt", "md", "py"]
+menu = ["htm", "txt", "xlsx", "csv", "md", "py"]  #619
 choice = st.sidebar.selectbox("Output File Type:", menu)
 model_choice = st.sidebar.radio("Select Model:", ('gpt-3.5-turbo', 'gpt-3.5-turbo-0301'))
 
@@ -91,6 +91,12 @@ def get_table_download_link(file_path):
     ext = os.path.splitext(file_name)[1]  # get the file extension
     if ext == '.txt':
         mime_type = 'text/plain'
+    elif ext == '.py':
+        mime_type = 'text/plain'
+    elif ext == '.xlsx':
+        mime_type = 'text/plain'
+    elif ext == '.csv':
+        mime_type = 'text/plain'
     elif ext == '.htm':
         mime_type = 'text/html'
     elif ext == '.md':
@@ -152,7 +158,7 @@ def main():
         #max_length = 12000 - optimal for gpt35 turbo. 2x=24000 for gpt4.  8x=96000 for gpt4-32k.
         max_length = st.slider("File section length for large files", min_value=1000, max_value=128000, value=12000, step=1000)
     with colupload:
-        uploaded_file = st.file_uploader("Add a file for context:", type=["xml", "json", "html", "htm", "md", "txt"])
+        uploaded_file = st.file_uploader("Add a file for context:", type=["xml", "json", "xlsx","csv","html", "htm", "md", "txt"])
     
     document_sections = deque()
     document_responses = {}