Update BrainBot.py

BrainBot.py

@@ -18,7 +18,7 @@ def save_uploaded_file(uploaded_file):
     
     # Create a directory if it doesn't exist
     data_dir = "/data"
-    os.makedirs(data_dir, exist_ok=True)
+    # 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:
@@ -30,7 +30,7 @@ def save_uploaded_file(uploaded_file):
 def save_uploaded_image(uploaded_image):
     # Create a directory named "images" if it doesn't exist
     images_dir = "/images"
-    os.makedirs(images_dir, exist_ok=True)
+    # os.makedirs(images_dir, exist_ok=True)
     
     # Create a temporary file path within the "images" directory with .png extension
     temp_file_path = os.path.join(images_dir, tempfile.NamedTemporaryFile(suffix=".png").name)
@@ -126,12 +126,12 @@ if uploaded_file is not None:
     with st.spinner("Loading file..."):
         # Save the uploaded file to a temporary path
         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}"
-            st.write(FASTAPI_URL)
             response = requests.post(FASTAPI_URL, json=data)
             st.success(response.text)
             st.session_state['current_file'] = uploaded_file.name