Update app.py

app.py

@@ -10,8 +10,6 @@ from langchain.prompts.chat import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.messages import HumanMessage, AIMessage
 from langchain.schema.output_parser import StrOutputParser
 
-import os
-
 st.set_page_config(page_title="ITC Financial Analysis ChatBot", layout="centered")
 
 # API key
@@ -19,7 +17,7 @@ API_key = os.environ.get("GOOGLE_API_KEY")
 if not API_key:
     st.error("❌ GOOGLE_API_KEY not found. Please set it in your Hugging Face Secrets.")
     st.stop()
-
+    
 st.markdown("""
 <h1 style='text-align: center; color: white; font-size: 42px;'>
 📊 ITC Financials with AI
@@ -61,10 +59,13 @@ st.markdown(new_chat_button, unsafe_allow_html=True)
 if st.button("New Chat"):
     memory_buffer["chat_history"] = []
 
+# import zipfile
+
 # Load Chroma vector DB from zip
-# with zipfile.ZipFile('database/chroma_db_32_backup.zip', 'r') as zip_ref:
+# with zipfile.ZipFile(r"C:\Users\hi\Desktop\Neuzen AI\chroma_db_backup.zip", 'r') as zip_ref:
 #     zip_ref.extractall('chroma_db')
 
+
 # Embeddings and vector store
 embedding = HuggingFaceEmbeddings(model_name='all-MiniLM-L6-v2')
 vectorstore = Chroma(persist_directory='database/chroma_db', embedding_function=embedding)
@@ -163,9 +164,10 @@ if user_input:
 
         # Show sources
         if output.get("source_documents"):
-            st.markdown("**Sources:**")
-            for doc in output["source_documents"]:
-                st.markdown(f"- {doc.metadata.get('source', 'Unknown document')}")
+            st.markdown("#### Source Documents", unsafe_allow_html=True)
+            for i, doc in enumerate(output["source_documents"], 1):
+                source_name = doc.metadata.get("source", f"Document {i}")
+                st.markdown(f"*{i}. {source_name}*")
 
 st.markdown(
     """
@@ -174,8 +176,7 @@ st.markdown(
             background-image: url("https://cdn-uploads.huggingface.co/production/uploads/673f5e166c2774fcc8a82f0b/i8t_a-SuqlO3MVVZv-Y0O.jpeg");
             background-size: cover;
             background-position: center;
-            background-repeat: no-repeat;
-            background-attachment: fixed;
+            height: 100vh;
         }
         
         /* Semi-transparent overlay */
@@ -191,4 +192,5 @@ st.markdown(
         }
     </style>
     """, 
-    unsafe_allow_html=True)
+    unsafe_allow_html=True
+)