Upload app.py

app.py

@@ -3,11 +3,16 @@ from src.file_loader import load_file
 from src.rag_pipeline import build_rag_pipeline, get_relevant_docs
 from src.model_utils import load_hf_model, generate_answer
 from src.utils import get_font_css
+from langchain.schema import Document
 
 st.set_page_config(page_title="AI Chatbot", page_icon=":robot_face:", layout="wide")
 st.markdown(get_font_css(), unsafe_allow_html=True)
 
-st.sidebar.image("assets/logo.png", width=180)
+try:
+    st.sidebar.image("assets/logo.png", width=180)
+except Exception:
+    st.sidebar.write("AI Chatbot")
+
 st.sidebar.title("AI Chatbot")
 st.sidebar.markdown("Upload a file to get started:")
 
@@ -38,7 +43,7 @@ st.markdown(
 if uploaded_file:
     with st.spinner("Processing file..."):
         text = load_file(uploaded_file)
-        docs = [{"page_content": chunk, "metadata": {}} for chunk in text]
+        docs = [Document(page_content=chunk, metadata={}) for chunk in text]
         retriever = build_rag_pipeline(docs, embedding_model)
         st.success("File processed and indexed!")
 
@@ -53,7 +58,7 @@ if uploaded_file:
     if st.button("Send", use_container_width=True) and user_input:
         with st.spinner("Generating answer..."):
             context_docs = get_relevant_docs(retriever, user_input)
-            context = " ".join([doc["page_content"] for doc in context_docs])
+            context = " ".join([doc.page_content for doc in context_docs])
             answer = generate_answer(text_gen, user_input, context)
             st.session_state.chat_history.append(("user", user_input))
             st.session_state.chat_history.append(("bot", answer))