Update app.py

app.py

@@ -95,7 +95,7 @@ def retrieve_relevant_docs(user_query, docs, index, bm25):
 
     return [docs[i] for i in combined_indices[:3]]
 
-def generate_response(user_query, company_ticker, mode, uploaded_file):
+def generate_response(user_query, pdf_ticker, ai_ticker, mode, uploaded_file):
     try:
         if mode == "📄 PDF Upload Mode":
             docs, embeddings, index, bm25 = extract_and_embed_text(uploaded_file)
@@ -104,11 +104,11 @@ def generate_response(user_query, company_ticker, mode, uploaded_file):
 
             retrieved_docs = retrieve_relevant_docs(user_query, docs, index, bm25)
             context = "\n\n".join(retrieved_docs)
-            prompt = f"Summarize the key financial insights from this document:\n\n{context}"
+            prompt = f"Summarize the key financial insights for {pdf_ticker} from this document:\n\n{context}"
 
         elif mode == "🌍 Live Data Mode":
-            financial_info = fetch_financial_data(company_ticker)
-            prompt = f"Analyze the financial status of {company_ticker} based on:\n{financial_info}\n\nUser Query: {user_query}"
+            financial_info = fetch_financial_data(ai_ticker)
+            prompt = f"Analyze the financial status of {ai_ticker} based on:\n{financial_info}\n\nUser Query: {user_query}"
         else:
             return "Invalid mode selected."
 
@@ -137,25 +137,29 @@ with col2:
     st.markdown("### 🔎 **Enter Your Query**")
     user_query = st.text_input("💬 What financial insights are you looking for?")
 
+st.markdown("---")
 if mode == "📄 PDF Upload Mode":
     st.markdown("### 📂 Upload Your Financial Report")
     uploaded_file = st.file_uploader("🔼 Upload PDF (Only for PDF Mode)", type=["pdf"])
-    company_ticker = None
+    pdf_ticker = st.text_input("🏢 Enter Company Ticker for PDF Insights", placeholder="e.g., INFY, TCS")
+    ai_ticker = None
 else:
     st.markdown("### 🌍 Live Market Data")
-    company_ticker = st.text_input("🏢 Enter Company Ticker Symbol", placeholder="e.g., AAPL, MSFT")
+    ai_ticker = st.text_input("🏢 Enter Company Ticker for AI Insights", placeholder="e.g., AAPL, MSFT")
     uploaded_file = None
+    pdf_ticker = None
 
 if st.button("🚀 Analyze Now"):
-    if mode == "📄 PDF Upload Mode" and not uploaded_file:
-        st.error("❌ Please upload a PDF file.")
-    elif mode == "🌍 Live Data Mode" and not company_ticker:
-        st.error("❌ Please enter a valid company ticker symbol.")
+    if mode == "📄 PDF Upload Mode" and (not uploaded_file or not pdf_ticker):
+        st.error("❌ Please upload a PDF and enter a company ticker for insights.")
+    elif mode == "🌍 Live Data Mode" and not ai_ticker:
+        st.error("❌ Please enter a valid company ticker for AI insights.")
     else:
         with st.spinner("🔍 Your Query is Processing, this can take up to 5 - 7 minutes ⏳"):
-            response = generate_response(user_query, company_ticker, mode, uploaded_file)
+            response = generate_response(user_query, pdf_ticker, ai_ticker, mode, uploaded_file)
             st.markdown("---")
             st.markdown("<h3 style='color: #4CAF50;'>💡 AI Response</h3>", unsafe_allow_html=True)
             st.write(response)
 
 st.markdown("---")
+