Update app.py

app.py

@@ -2,9 +2,9 @@ import streamlit as st
 import pandas as pd
 import json
 import os
-from VisionaryAgent import search_company, scrape_website, process_company_description, process_uploaded_document
-from VisionaryAgent import get_industry_trends, get_ai_use_cases, get_competitor_ai_strategies
-from VisionaryAgent import generate_ai_strategy, suggest_ai_integration, identify_revenue_opportunities, generate_report
+from phase1_agents import search_company, scrape_website, process_company_description, process_uploaded_document
+from phase2_agents import get_industry_trends, get_ai_use_cases, get_competitor_ai_strategies
+from phase3_agents import generate_ai_strategy, suggest_ai_integration, identify_revenue_opportunities, generate_report
 
 # Define data storage paths
 CSV_FILE = "user_data.csv"
@@ -35,6 +35,23 @@ def main():
     st.title("Visionary AI  by Giant Analytics")
     st.write("Fill in the details to generate an AI-driven business strategy report.")
     st.write("It uses SOTA (State-of-the-Art) Reasoning Models to provide cutting-edge insights and AI integration strategies.")
+    
+    # Initialize session state for data persistence
+    if "company_data" not in st.session_state:
+        st.session_state.company_data = ""
+    if "industry_trends" not in st.session_state:
+        st.session_state.industry_trends = ""
+    if "ai_use_cases" not in st.session_state:
+        st.session_state.ai_use_cases = ""
+    if "competitor_analysis" not in st.session_state:
+        st.session_state.competitor_analysis = ""
+    if "ai_strategy" not in st.session_state:
+        st.session_state.ai_strategy = ""
+    if "ai_integration" not in st.session_state:
+        st.session_state.ai_integration = ""
+    if "revenue_opportunities" not in st.session_state:
+        st.session_state.revenue_opportunities = ""
+    
     # Collect User Information
     name = st.text_input("Name")
     email = st.text_input("Email")
@@ -45,73 +62,93 @@ def main():
     input_method = st.radio("How would you like to provide company details?", 
                             ("Search by Name", "Website URL", "Manual Description", "Upload Document"))
     
-    company_data = ""
     if input_method == "Search by Name":
         if st.button("Find Company Details"):
-            company_data = search_company(company_name)
-            st.write(company_data)
+            st.session_state.company_data = search_company(company_name)
     elif input_method == "Website URL":
         website_url = st.text_input("Enter Website URL")
         if st.button("Scrape Website"):
-            company_data = scrape_website(website_url)
-            st.write(company_data)
+            st.session_state.company_data = scrape_website(website_url)
     elif input_method == "Manual Description":
-        company_data = st.text_area("Enter Company Description")
+        company_data_input = st.text_area("Enter Company Description")
         if st.button("Process Description"):
-            company_data = process_company_description(company_data)
-            st.write(company_data)
+            st.session_state.company_data = process_company_description(company_data_input)
     elif input_method == "Upload Document":
         uploaded_file = st.file_uploader("Upload PDF or PPT", type=["pdf", "pptx"])
         if uploaded_file is not None:
-            company_data = process_uploaded_document(uploaded_file)
-            st.write(company_data)
+            st.session_state.company_data = process_uploaded_document(uploaded_file)
     
-    if company_data:
+    if st.session_state.company_data:
+        st.subheader("Extracted Company Information")
+        st.write(st.session_state.company_data)
+        
         industry = st.text_input("Industry Type (e.g., Healthcare, Finance)")
         if st.button("Analyze Industry Trends"):
-            industry_trends = get_industry_trends(industry)
-            st.write(industry_trends)
+            st.session_state.industry_trends = get_industry_trends(industry)
         
+        if st.session_state.industry_trends:
+            st.subheader("Industry Trends")
+            st.write(st.session_state.industry_trends)
+            
         if st.button("Find AI Use Cases"):
-            ai_use_cases = get_ai_use_cases(industry)
-            st.write(ai_use_cases)
+            st.session_state.ai_use_cases = get_ai_use_cases(industry)
         
+        if st.session_state.ai_use_cases:
+            st.subheader("AI Use Cases")
+            st.write(st.session_state.ai_use_cases)
+            
         competitor = st.text_input("Enter Competitor Name")
         if st.button("Analyze Competitor AI Strategies"):
-            competitor_analysis = get_competitor_ai_strategies(competitor)
-            st.write(competitor_analysis)
+            st.session_state.competitor_analysis = get_competitor_ai_strategies(competitor)
+        
+        if st.session_state.competitor_analysis:
+            st.subheader("Competitor AI Strategies")
+            st.write(st.session_state.competitor_analysis)
         
         if st.button("Generate AI Strategy"):
-            ai_strategy = generate_ai_strategy(company_data, industry_trends, ai_use_cases, competitor_analysis)
-            st.write(ai_strategy)
+            st.session_state.ai_strategy = generate_ai_strategy(
+                st.session_state.company_data, st.session_state.industry_trends,
+                st.session_state.ai_use_cases, st.session_state.competitor_analysis)
         
+        if st.session_state.ai_strategy:
+            st.subheader("AI Strategy")
+            st.write(st.session_state.ai_strategy)
+            
         if st.button("Suggest AI Integration Plan"):
-            ai_integration = suggest_ai_integration(company_data, ai_strategy)
-            st.write(ai_integration)
+            st.session_state.ai_integration = suggest_ai_integration(
+                st.session_state.company_data, st.session_state.ai_strategy)
         
+        if st.session_state.ai_integration:
+            st.subheader("AI Integration Plan")
+            st.write(st.session_state.ai_integration)
+            
         if st.button("Identify Revenue Growth Opportunities"):
-            revenue_opportunities = identify_revenue_opportunities(company_data, ai_strategy)
-            st.write(revenue_opportunities)
+            st.session_state.revenue_opportunities = identify_revenue_opportunities(
+                st.session_state.company_data, st.session_state.ai_strategy)
         
+        if st.session_state.revenue_opportunities:
+            st.subheader("Revenue Growth Opportunities")
+            st.write(st.session_state.revenue_opportunities)
+            
         if st.button("Generate Final Report"):
-            report_filename = generate_report(company_name, ai_strategy, ai_integration, revenue_opportunities)
+            report_filename = generate_report(
+                company_name, st.session_state.ai_strategy, st.session_state.ai_integration, 
+                st.session_state.revenue_opportunities)
             st.success(f"Report Generated: {report_filename}")
-            
+        
         # Save data to backend
         user_data = {
-            "name": name,
-            "email": email,
-            "mobile": mobile,
-            "company_name": company_name,
-            "company_data": company_data,
-            "industry": industry,
-            "competitor": competitor,
-            "ai_strategy": ai_strategy,
-            "ai_integration": ai_integration,
-            "revenue_opportunities": revenue_opportunities
+            "name": name, "email": email, "mobile": mobile, "company_name": company_name,
+            "company_data": st.session_state.company_data, "industry": industry,
+            "competitor": competitor, "ai_strategy": st.session_state.ai_strategy,
+            "ai_integration": st.session_state.ai_integration,
+            "revenue_opportunities": st.session_state.revenue_opportunities
         }
         save_data_csv(user_data)
         save_data_json(user_data)
 
+st.markdown("---")
+st.markdown("**Developed by [Aditya Ghadge](https://www.linkedin.com/in/aditya-ghadge-a82b30240/) for Giant Analytics**")
+
 if __name__ == "__main__":
-    main()
+    main()