Update BrainBot.py

BrainBot.py

@@ -14,19 +14,18 @@ st.markdown(f'<style>{css}</style>', unsafe_allow_html=True)
 ## -------------------------------------------------------------------------------------------
 # Function to save the uploaded file as a temporary file and return its path.
 def save_uploaded_file(uploaded_file):
-    st.write("Inside func")
-    # file_content = uploaded_file.read()  # Load the document
-    # st.write("read file")
-    # # Create a directory if it doesn't exist
-    # data_dir = "/data"
-    # # os.makedirs(data_dir, exist_ok=True)
+    file_content = uploaded_file.read()  # Load the document
+    st.write("read file")
+    # Create a directory if it doesn't exist
+    data_dir = "/data"
+    # os.makedirs(data_dir, exist_ok=True)
     
-    # # Create a temporary file in the data directory
-    # with tempfile.NamedTemporaryFile(delete=False, dir=data_dir) as temp_file:
-    #     st.write("writing file...")
-    #     temp_file.write(file_content)  # Write the uploaded file content to the temporary file
-    #     temp_file_path = temp_file.name  # Get the path of the temporary file
-    #     return temp_file_path
+    # Create a temporary file in the data directory
+    with tempfile.NamedTemporaryFile(delete=False, dir=data_dir) as temp_file:
+        st.write("writing file...")
+        temp_file.write(file_content)  # Write the uploaded file content to the temporary file
+        temp_file_path = temp_file.name  # Get the path of the temporary file
+        return temp_file_path
 
 # Function to save the uploaded image as a temporary file  and return its path.
 def save_uploaded_image(uploaded_image):
@@ -130,17 +129,17 @@ if uploaded_file is not None:
         temp_file_path = save_uploaded_file(uploaded_file)
         st.write(temp_file_path)
             
-        # try:    
-        #     # Send POST request to a FastAPI endpoint to load the file into a vectorstore
-        #     data = {"file_path": temp_file_path, "file_type": uploaded_file.type}
-        #     FASTAPI_URL = f"http://localhost:7860/load_file/{llm}"
-        #     response = requests.post(FASTAPI_URL, json=data)
-        #     st.success(response.text)
-        #     st.session_state['current_file'] = uploaded_file.name
-        #     st.session_state['uploaded_file'] = True
-        #     st.switch_page("pages/File-chat.py")
-        # except Exception as e:
-        #     st.switch_page("pages/error.py")
+        try:    
+            # Send POST request to a FastAPI endpoint to load the file into a vectorstore
+            data = {"file_path": temp_file_path, "file_type": uploaded_file.type}
+            FASTAPI_URL = f"http://localhost:7860/load_file/{llm}"
+            response = requests.post(FASTAPI_URL, json=data)
+            st.success(response.text)
+            st.session_state['current_file'] = uploaded_file.name
+            st.session_state['uploaded_file'] = True
+            st.switch_page("pages/File-chat.py")
+        except Exception as e:
+            st.switch_page("pages/error.py")
 
 ## IMAGE
 ## -------------------------------------------------------------------------------------------