Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,47 +2,69 @@ import streamlit as st
 import os
 from rag_engine import ProjectRAGEngine
 
-st.set_page_config(page_title="Project Analyzer", layout="wide")
+# Page configuration to ensure wide display for citations [cite: 31]
+st.set_page_config(page_title="IIR Project Analyzer", layout="wide")
 st.title("📂 Industrial Project Report Analyzer")
 
-# Handling OpenAI API Key from Hugging Face Secrets or User Input
+# 1. API Key Security Logic [cite: 21]
+# Checks Hugging Face Secrets first, then falls back to user input
 api_key = os.environ.get("OPENAI_API_KEY")
 if not api_key:
     api_key = st.sidebar.text_input("Enter OpenAI API Key", type="password")
 
 if not api_key:
-    st.warning("Please provide an OpenAI API Key to continue.")
+    st.warning("Please provide an OpenAI API Key in the sidebar to proceed.")
     st.stop()
 
+# 2. Session Persistence [cite: 8, 44]
 if "engine" not in st.session_state:
     st.session_state.engine = ProjectRAGEngine(api_key)
 
-# Document Upload Section [cite: 29]
-uploaded_files = st.sidebar.file_uploader("Upload Project PDFs", type="pdf", accept_multiple_files=True)
+# 3. Document Management Interface [cite: 6, 29]
+uploaded_files = st.sidebar.file_uploader(
+    "Upload Project PDFs", 
+    type="pdf", 
+    accept_multiple_files=True
+)
 
 if uploaded_files:
+    # Check if files are already processed to prevent redundant API calls [cite: 8]
     if "processed_files" not in st.session_state or len(st.session_state.processed_files) != len(uploaded_files):
-        with st.spinner("Processing documents..."):
-            if not os.path.exists("temp"): os.makedirs("temp")
+        with st.spinner("Analyzing project reports..."):
+            # Ensure the temp directory has write permissions to avoid 403/Permission errors
+            temp_dir = "temp"
+            if not os.path.exists(temp_dir): 
+                os.makedirs(temp_dir, mode=0o777)
+            
             paths = []
             for f in uploaded_files:
-                p = os.path.join("temp", f.name)
-                with open(p, "wb") as out: out.write(f.getbuffer())
-                paths.append(p)
+                path = os.path.join(temp_dir, f.name)
+                with open(path, "wb") as out:
+                    out.write(f.getbuffer())
+                paths.append(path)
+            
+            # Simultaneous querying setup [cite: 7]
             st.session_state.engine.process_documents(paths)
             st.session_state.processed_files = [f.name for f in uploaded_files]
-        st.success("Indexing complete. You can now ask questions.")
+        st.success("Reports indexed. Ready for queries.")
 
-# Chat Interface [cite: 30, 31]
+# 4. Question Answering Interface [cite: 30-31]
 query = st.chat_input("Ex: 'Compare the budgets of these projects'")
+
 if query:
-    with st.chat_message("user"): st.write(query)
+    with st.chat_message("user"):
+        st.write(query)
+
     with st.chat_message("assistant"):
+        # Handle cases with no documents [cite: 25]
         answer, sources = st.session_state.engine.get_answer(query)
-        st.write(answer)
-        with st.expander("Source Attribution & Quotes"):
-            for s in sources:
-                doc = os.path.basename(s['metadata']['source'])
-                page = s['metadata']['page']
-                st.markdown(f"**{doc} (Page {page})**")
-                st.caption(f"\"{s['content'][:200]}...\"")
+        st.markdown(f"### Response\n{answer}") # cite: 16
+        
+        # Mandatory Source Attribution Display [cite: 11, 17, 31]
+        if sources:
+            with st.expander("View Source Attribution & Quotes"): # cite: 18
+                for idx, s in enumerate(sources):
+                    doc_name = os.path.basename(s['metadata']['source'])
+                    page_num = s['metadata']['page'] + 1 # PDF index starts at 0
+                    st.markdown(f"**Source {idx+1}:** {doc_name} (Page {page_num})")
+                    st.caption(f"Verbatim Quote: \"{s['content'][:300]}...\"")