Update app.py

app.py

@@ -1,38 +1,50 @@
 import streamlit as st
-
-# Set page configuration first
-st.set_page_config(page_title="FinBrief: Financial Document Insights", layout="wide")
-
 import spacy
 import pandas as pd
 import re
-from transformers import pipeline
 import subprocess
+from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
+
+st.set_page_config(page_title="FinBrief: Financial Document Insights", layout="wide")
 
-# Load the spaCy model once at the start
+# Debugging print
+print("===== Starting FinBrief App =====")
+
+# Load spaCy model
 try:
     nlp = spacy.load("en_core_web_sm")
-    st.write("spaCy model loaded successfully!")
+    print("spaCy model loaded successfully!")
 except OSError:
-    st.write("Failed to load spaCy model.")
+    print("❌ spaCy model not found. Attempting to install...")
     subprocess.run(["python", "-m", "spacy", "download", "en_core_web_sm"])
-    nlp = spacy.load("en_core_web_sm")
-
-# Load the summarization model from Hugging Face Model Hub
+    try:
+        nlp = spacy.load("en_core_web_sm")
+        print("spaCy model installed and loaded successfully!")
+    except Exception as e:
+        print("❌ Failed to load spaCy model after installation:", e)
+        nlp = None
+
+# Load Hugging Face model
+model_name = "kritsadaK/bart-financial-summarization"
 try:
-    online_model_path = "kritsadaK/bart-financial-summarization"
-    summarizer = pipeline("summarization", model=online_model_path, tokenizer=online_model_path)
-    st.write("Online summarization model loaded successfully!")
+    print(f"Attempting to load Hugging Face model: {model_name}")
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+    summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
+    print("Hugging Face model loaded successfully!")
 except Exception as e:
-    summarizer = None  # Handle case where model is missing
-    st.write("Failed to load online summarization model.")
+    print(f"Failed to load Hugging Face model: {e}")
+    summarizer = None
 
-# Initialize models in session state if not already loaded
+# Initialize session state
 if "nlp" not in st.session_state:
     st.session_state["nlp"] = nlp
 if "summarizer" not in st.session_state:
     st.session_state["summarizer"] = summarizer
 
+# Debugging: Check session state initialization
+print(f"Session State - NLP: {st.session_state['nlp'] is not None}, Summarizer: {st.session_state['summarizer'] is not None}")
+
 # # Load the summarization model locally
 # try:
 #     local_model_path = "./local_models/bart-financial"