modified dataloader_page

src/backend/routes/upsert_data.py

@@ -13,7 +13,7 @@ async def upsert_data(add_data: Add_Data_In_DB):
         start = add_data.start
         end = add_data.end
         df = dataset.get_data_set()[start:end]
-        pinecone_db.upsert_data_in_db(df, index_name)
+        pinecone_db.upsert_data_in_db(df)
         return {"status": "success"}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))

src/frontend/app/common_fuctions.py

@@ -45,7 +45,7 @@ def upsert_data_request(start, end):
     try:
         url = "http://localhost:8000/data/upsert_data"
         response = requests.post(url, data=json.dumps(payload), headers=headers)
-        return response
+        return response.json()
     except requests.exceptions.HTTPError as http_err:
         print(f"HTTP error occurred: {http_err}")
     except Exception as err:

src/frontend/pages/dataloader_page.py

@@ -27,14 +27,10 @@ def load_data():
 
             with st.spinner("⏳ Upserting data... Please wait"):
                 response = common_fuctions.upsert_data_request(start_index, end_index)
-                st.write(response)
-                # if response.get("status") == "success":
-                #     st.success("Data upserted successfully!")
-                #     st.session_state.load_clicked = False
-                # else:
-                #     st.error("Failed to upsert data.")
-                #     logger.error("Failed to upsert data.")
-                #     st.session_state.load_clicked = False
+                if response.get("status") == "success": 
+                    st.success("Data upserted successfully!")
+                else:
+                    st.error("Error upserting data.")
     except Exception as e:
         st.error(f"Error loading data: {e}")
         logger.error(f"Error loading data: {e}")