Final app.py

app.py

@@ -3,10 +3,6 @@ os.environ["STREAMLIT_SERVER_PORT"] = "8501"
 os.environ["STREAMLIT_SERVER_HEADLESS"] = "true"
 
 import streamlit as st
-st.set_page_config(page_title="DocAnalyzer Pro", layout="wide")
-
-st.write("Step 1: Imports done")
-
 from transformers import pipeline
 from PyPDF2 import PdfReader
 import docx
@@ -15,13 +11,34 @@ import psutil
 from pathlib import Path
 import torch
 
-st.write("Step 2: Libraries imported")
+# Page config with wide layout
+st.set_page_config(page_title="LexPilot", layout="wide")
+
+# Sidebar with project info
+with st.sidebar:
+    st.title("LexPilot™")
+    st.markdown(
+        """
+        LexPilot™ ingests text, PDF, and Word files to instantly analyze contracts.  
+        It delivers concise summaries and lets you ask targeted questions—  
+        giving fast, precise insights to speed up legal and procurement reviews.
+        """
+    )
+    st.markdown("---")
+    st.write("### System Status")
+    try:
+        device_status = 'GPU ✅' if torch.cuda.is_available() else 'CPU ⚠️'
+    except:
+        device_status = 'CPU (torch not configured)'
+    st.text(f"Device: {device_status}")
+    st.text(f"Memory: {psutil.virtual_memory().percent}% used")
+    st.text(f"CPU: {psutil.cpu_percent()}% used")
 
+# Setup cache directory for models
 def setup_environment():
     cache_dir = Path(".cache/models")
     try:
         cache_dir.mkdir(exist_ok=True, parents=True)
-        st.write(f"Step 3: Cache directory setup at {cache_dir}")
     except Exception as e:
         st.error(f"Failed to create cache directory: {e}")
     return cache_dir
@@ -31,23 +48,19 @@ cache_dir = setup_environment()
 @st.cache_resource(ttl=3600)
 def load_models():
     try:
-        st.info("Loading QA model...")
         qa_model = pipeline(
             "question-answering",
             model="distilbert-base-cased-distilled-squad",
             device=-1
         )
-        st.info("QA model loaded.")
-        st.info("Loading summarizer model...")
         summarizer_model = pipeline(
             "summarization",
-            model="sshleifer/distilbart-cnn-6-6",  # smaller model for faster startup
+            model="sshleifer/distilbart-cnn-6-6",
             device=-1
         )
-        st.info("Summarizer model loaded.")
         return {'qa': qa_model, 'summarizer': summarizer_model}
     except Exception as e:
-        st.error(f"❌ Failed to load models: {e}")
+        st.error(f"Failed to load models: {e}")
         st.stop()
 
 models = load_models()
@@ -88,8 +101,8 @@ def generate_summary(text, max_length=150):
         st.error(f"Summarization failed: {e}")
         return ""
 
-st.title("📄 DocAnalyzer Pro")
-st.write("Step 4: UI started")
+# Main UI title
+st.title("📄 LexPilot")
 
 with st.expander("📤 Upload Document", expanded=True):
     uploaded_file = st.file_uploader("Choose PDF/DOCX", type=["pdf", "docx"])
@@ -128,15 +141,7 @@ with tab2:
                         st.success(f"Generated in {time.time()-start_time:.1f}s")
                         st.markdown(f"**Summary:**\n\n{summary}")
 
-with st.expander("⚙️ System Status"):
-    try:
-        device_status = 'GPU ✅' if torch.cuda.is_available() else 'CPU ⚠️'
-    except:
-        device_status = 'CPU (torch not configured)'
-    st.code(f"""
-Models loaded: {', '.join(models.keys())}
-Device: {device_status}
-Memory: {psutil.virtual_memory().percent}% used
-CPU: {psutil.cpu_percent()}% used
-Cache location: {cache_dir}
-""")
+# Show cache dir path in sidebar (optional)
+with st.sidebar:
+    st.markdown("---")
+    st.write(f"Cache directory: {cache_dir}")