Update BrainBot.py

BrainBot.py

@@ -8,19 +8,7 @@ import os
 with open('styles.css') as f:
     css = f.read()
 
-st.markdown(f'<style>{css}</style>', unsafe_allow_html=True)
-
-## SERVER SETUP
-## -------------------------------------------------------------------------------------------
-# def start_server():   
-#     os.system("uvicorn main:app --port 7860 --host 0.0.0.0 --workers 2")
-#     st.session_state['server_started'] = True
- 
-# if 'server_started' not in st.session_state:
-#     st.session_state['server_started'] = False
-
-# if not st.session_state['server_started']:
-#   start_server()              
+st.markdown(f'<style>{css}</style>', unsafe_allow_html=True)           
 
 ## FUNCTIONS
 ## -------------------------------------------------------------------------------------------
@@ -29,7 +17,7 @@ def save_uploaded_file(uploaded_file):
     file_content = uploaded_file.read()  # Load the document
     
     # Create a directory if it doesn't exist
-    data_dir = "data"
+    data_dir = "/data"
     # os.makedirs(data_dir, exist_ok=True)
     
     # Create a temporary file in the data directory
@@ -93,14 +81,13 @@ with col3:
         # variable
         with st.spinner("Activating OpenAI API..."):
             try:
-                # FASTAPI_URL = "http://localhost:8000/set_api_key"
                 FASTAPI_URL = "http://localhost:7860/set_api_key"
                 data = {"api_key": openai_api_key}
                 if openai_api_key:
                     response = requests.post(FASTAPI_URL, json=data)
                     st.sidebar.success(response.text)
                     st.session_state['api_key_flag'] = True
-                    # st.experimental_rerun()
+                    st.experimental_rerun()
             except Exception as e:
                 st.switch_page("pages/error.py")
 with col4:
@@ -143,7 +130,7 @@ if uploaded_file is not None:
         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:8000/load_file/{llm}"
+            FASTAPI_URL = f"http://localhost:7860/load_file/{llm}"
             st.write(FASTAPI_URL)
             response = requests.post(FASTAPI_URL, json=data)
             st.success(response.text)
@@ -181,7 +168,7 @@ if website_link is not None:
             try:
                 # Send POST request to a FastAPI endpoint to scrape the webpage and load its text 
                 # into a vector store
-                FASTAPI_URL = f"http://localhost:8000/load_link/{llm}"
+                FASTAPI_URL = f"http://localhost:7860/load_link/{llm}"
                 data = {"website_link": website_link}
                 with st.spinner("Loading website..."):
                     response = requests.post(FASTAPI_URL, json=data)