Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,8 +7,11 @@ import logging
 import time
 import os
 from dotenv import load_dotenv
-from langchain_mymod.langgraph_init.UploadDocs import upload_documents
-from langchain_mymod.langgraph_init import submit_feedback, get_conversations,get_conversation_history, chat_with_rag, login_hf
+from langgraph_init import UploadDocs
+from langgraph_init import submit_feedback, get_conversations,get_conversation_history, chat_with_rag, login_hf
+
+upd = UploadDocs()
+
 # --- 1. Configuration ---
 # API_URL = "https://sra25-fastapi-backend.hf.space"
 # API_URL = "http://localhost:8000"
@@ -173,7 +176,7 @@ with st.sidebar:
             if new_uploaded_files:
                 st.session_state.uploaded_files_info.extend(newly_added_files_info)
                 with st.spinner("Processing documents..."):
-                    process_status = upload_documents(new_uploaded_files)
+                    process_status = upd.upload_documents(new_uploaded_files)
                     if process_status:
                         st.session_state.retriever_ready = True
                         # st.success(response_data.get("message", "Documents processed and knowledge base ready!"))
@@ -186,7 +189,7 @@ with st.sidebar:
             elif duplicate_file:
                 # st.session_state.uploaded_files_info.extend(newly_added_files_info)
                 with st.spinner("Processing documents..."):
-                    process_status = upload_documents(duplicate_file)
+                    process_status = upd.upload_documents(duplicate_file)
                     if process_status:
                         st.session_state.retriever_ready = True
                         # st.success(response_data.get("message", "Documents processed and knowledge base ready!"))