Update app.py

app.py

@@ -1,186 +1,30 @@
 import streamlit as st
-import os
-import json
 import pandas as pd
-from phase1_agents import search_company, scrape_website, process_company_description
-from phase2_agents import get_industry_trends, get_ai_use_cases
-from phase3_agents import generate_ai_strategy, suggest_ai_integration, identify_revenue_opportunities, generate_report
-from bni_agent import get_bni_benefits
-from rag_agent import recommend_bni_connections
-
-# File Paths for Storing User Data
-CSV_FILE = "user_data.csv"
-JSON_FILE = "user_data.json"
-
-# Function to Save Data to CSV
-def save_data_csv(data):
-    df = pd.DataFrame([data])
-    if os.path.exists(CSV_FILE):
-        df.to_csv(CSV_FILE, mode='a', header=False, index=False)
-    else:
-        df.to_csv(CSV_FILE, index=False)
-
-# Function to Save Data to JSON
-def save_data_json(data):
-    if os.path.exists(JSON_FILE):
-        with open(JSON_FILE, "r") as file:
-            existing_data = json.load(file)
-    else:
-        existing_data = []
-    
-    existing_data.append(data)
-    with open(JSON_FILE, "w") as file:
-        json.dump(existing_data, file, indent=4)
-
-# Streamlit UI
-def main():
-    st.title("🚀 Visionary AI - AI Strategy Generator")
-    st.write("📊 This tool provides an AI-driven business strategy report tailored for your company.")
-    st.write("🔍 Powered by **State-of-the-Art Reasoning Models** for AI integration insights.")
-
-    # Initialize Session State Variables
-    session_keys = ["company_data", "industry_trends", "ai_use_cases", "ai_strategy", 
-                    "ai_integration", "revenue_opportunities", "bni_benefits", "bni_connections"]
-    
-    for key in session_keys:
-        if key not in st.session_state:
-            st.session_state[key] = None
-
-    # Collect User Information
-    name = st.text_input("👤 Your Name")
-    email = st.text_input("📧 Email")
-    mobile = st.text_input("📱 Mobile Number")
-    company_name = st.text_input("🏢 Company Name")
-    company_industry = st.text_input("💼 Industry (e.g., Healthcare, Finance, IT)")
-
-    # Select Company Data Input Method
-    st.subheader("🌐 Provide Company Information")
-    input_method = st.radio("Choose a method to provide company details:", 
-                            ["Company Website", "Web Search", "Manual Description"])
-
-    progress_bar = st.progress(0)
-
-    # Handle Company Data Input
-    if input_method == "Company Website":
-        website_url = st.text_input("🔗 Enter Company Website URL")
-        if st.button("🔍 Scrape Website"):
-            progress_bar.progress(10)
-            st.session_state.company_data = scrape_website(website_url)
-            progress_bar.progress(30)
-
-    elif input_method == "Web Search":
-        if st.button("🔍 Search Company Online"):
-            progress_bar.progress(10)
-            st.session_state.company_data = search_company(company_name)
-            progress_bar.progress(30)
-
-    elif input_method == "Manual Description":
-        company_description = st.text_area("📝 Enter Company Description")
-        if st.button("✔ Process Description"):
-            progress_bar.progress(10)
-            st.session_state.company_data = process_company_description(company_description)
-            progress_bar.progress(30)
-
-    # Automated Processing of All Phases
-    if st.session_state.company_data:
-        st.subheader("✅ Company Information Extracted")
-        st.write(st.session_state.company_data)
-        progress_bar.progress(40)
-
-        # Industry Analysis
-        if st.button("🔬 Analyze Industry & AI Trends"):
-            progress_bar.progress(50)
-            st.session_state.industry_trends = get_industry_trends(company_industry)
-            st.session_state.ai_use_cases = get_ai_use_cases(company_industry)
-            progress_bar.progress(60)
-
-        if st.session_state.industry_trends:
-            st.subheader("📈 Industry Trends & AI Use Cases")
-            st.write(st.session_state.industry_trends)
-            st.write(st.session_state.ai_use_cases)
-            progress_bar.progress(70)
-
-        # AI Strategy & Implementation
-        if st.button("🤖 Generate AI Strategy"):
-            progress_bar.progress(75)
-            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.ai_integration = suggest_ai_integration(st.session_state.company_data, 
-                                                                     st.session_state.ai_strategy)
-            st.session_state.revenue_opportunities = identify_revenue_opportunities(st.session_state.company_data, 
-                                                                                     st.session_state.ai_strategy)
-            progress_bar.progress(85)
-
-        if st.session_state.ai_strategy:
-            st.subheader("🚀 AI Strategy & Implementation Plan")
-            st.write(st.session_state.ai_strategy)
-            st.write(st.session_state.ai_integration)
-            st.write(st.session_state.revenue_opportunities)
-            progress_bar.progress(90)
-
-        # BNI Benefits & Recommendations
-        if st.button("🔗 Discover BNI Benefits"):
-            progress_bar.progress(92)
-            st.session_state.bni_benefits = get_bni_benefits(st.session_state.company_data)
-            st.session_state.bni_connections = recommend_bni_connections(st.session_state.company_data)
-            progress_bar.progress(95)
-
-        if st.session_state.bni_benefits:
-            st.subheader("🌍 BNI Membership Benefits for Your Business")
-            st.write(st.session_state.bni_benefits)
-
-        if st.session_state.bni_connections:
-            st.subheader("🤝 Recommended BNI Pearl Chapter Connections")
-            st.write(st.session_state.bni_connections)
-            progress_bar.progress(98)
-
-        # Generate Final Report
-        if st.button("📄 Generate Final Report"):
-            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}**")
-            progress_bar.progress(100)
-
-    # Footer
-    st.markdown("---")
-    st.markdown("🚀 **Developed by [Aditya Ghadge](https://www.linkedin.com/in/aditya-ghadge-a82b30240/) for Giant Analytics**")
-
-if __name__ == "__main__":
-    main()
-import streamlit as st
 import os
-import json
-import pandas as pd
 from phase1_agents import search_company, scrape_website, process_company_description
 from phase2_agents import get_industry_trends, get_ai_use_cases
 from phase3_agents import generate_ai_strategy, suggest_ai_integration, identify_revenue_opportunities, generate_report
 from bni_agent import get_bni_benefits
 from rag_agent import recommend_bni_connections
 
-# File Paths for Storing User Data
-CSV_FILE = "user_data.csv"
-JSON_FILE = "user_data.json"
-
-# Function to Save Data to CSV
-def save_data_csv(data):
-    df = pd.DataFrame([data])
-    if os.path.exists(CSV_FILE):
-        df.to_csv(CSV_FILE, mode='a', header=False, index=False)
-    else:
-        df.to_csv(CSV_FILE, index=False)
-
-# Function to Save Data to JSON
-def save_data_json(data):
-    if os.path.exists(JSON_FILE):
-        with open(JSON_FILE, "r") as file:
-            existing_data = json.load(file)
-    else:
-        existing_data = []
-    
-    existing_data.append(data)
-    with open(JSON_FILE, "w") as file:
-        json.dump(existing_data, file, indent=4)
+from datasets import load_dataset, Dataset
+from huggingface_hub import HfApi
+
+# Hugging Face Dataset Name (Replace with your actual dataset)
+HF_DATASET_NAME = "GiantAnalytics/visionary-ai-user-data"
+
+# Function to Save Data to Hugging Face Dataset
+def save_data_hf(data):
+    try:
+        dataset = load_dataset(HF_DATASET_NAME)
+        df = pd.DataFrame(dataset["train"])
+        new_df = pd.DataFrame([data])
+        df = pd.concat([df, new_df], ignore_index=True)
+        new_dataset = Dataset.from_pandas(df)
+        new_dataset.push_to_hub(HF_DATASET_NAME, split="train")
+        print("✅ Data saved successfully to Hugging Face Dataset")
+    except Exception as e:
+        print(f"❌ Failed to save data: {e}")
 
 # Streamlit UI
 def main():
@@ -191,7 +35,7 @@ def main():
     # Initialize Session State Variables
     session_keys = ["company_data", "industry_trends", "ai_use_cases", "ai_strategy", 
                     "ai_integration", "revenue_opportunities", "bni_benefits", "bni_connections"]
-    
+
     for key in session_keys:
         if key not in st.session_state:
             st.session_state[key] = None
@@ -231,18 +75,17 @@ def main():
             st.session_state.company_data = process_company_description(company_description)
             progress_bar.progress(30)
 
-    # Automated Processing of All Phases
+    # **Execute All Phases Automatically After Company Data is Available**
     if st.session_state.company_data:
         st.subheader("✅ Company Information Extracted")
         st.write(st.session_state.company_data)
         progress_bar.progress(40)
 
         # Industry Analysis
-        if st.button("🔬 Analyze Industry & AI Trends"):
-            progress_bar.progress(50)
-            st.session_state.industry_trends = get_industry_trends(company_industry)
-            st.session_state.ai_use_cases = get_ai_use_cases(company_industry)
-            progress_bar.progress(60)
+        progress_bar.progress(50)
+        st.session_state.industry_trends = get_industry_trends(company_industry)
+        st.session_state.ai_use_cases = get_ai_use_cases(company_industry)
+        progress_bar.progress(60)
 
         if st.session_state.industry_trends:
             st.subheader("📈 Industry Trends & AI Use Cases")
@@ -251,16 +94,15 @@ def main():
             progress_bar.progress(70)
 
         # AI Strategy & Implementation
-        if st.button("🤖 Generate AI Strategy"):
-            progress_bar.progress(75)
-            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.ai_integration = suggest_ai_integration(st.session_state.company_data, 
-                                                                     st.session_state.ai_strategy)
-            st.session_state.revenue_opportunities = identify_revenue_opportunities(st.session_state.company_data, 
-                                                                                     st.session_state.ai_strategy)
-            progress_bar.progress(85)
+        progress_bar.progress(75)
+        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.ai_integration = suggest_ai_integration(st.session_state.company_data, 
+                                                                 st.session_state.ai_strategy)
+        st.session_state.revenue_opportunities = identify_revenue_opportunities(st.session_state.company_data, 
+                                                                                 st.session_state.ai_strategy)
+        progress_bar.progress(85)
 
         if st.session_state.ai_strategy:
             st.subheader("🚀 AI Strategy & Implementation Plan")
@@ -269,12 +111,15 @@ def main():
             st.write(st.session_state.revenue_opportunities)
             progress_bar.progress(90)
 
-        # BNI Benefits & Recommendations
-        if st.button("🔗 Discover BNI Benefits"):
-            progress_bar.progress(92)
+        # **BNI Benefits & Recommendations (Handled with Error Handling)**
+        try:
             st.session_state.bni_benefits = get_bni_benefits(st.session_state.company_data)
             st.session_state.bni_connections = recommend_bni_connections(st.session_state.company_data)
             progress_bar.progress(95)
+        except Exception as e:
+            print(f"❌ Error fetching BNI data: {e}")
+            st.session_state.bni_benefits = "No BNI data available."
+            st.session_state.bni_connections = "No relevant BNI connections found."
 
         if st.session_state.bni_benefits:
             st.subheader("🌍 BNI Membership Benefits for Your Business")
@@ -285,16 +130,26 @@ def main():
             st.write(st.session_state.bni_connections)
             progress_bar.progress(98)
 
-        # Generate Final Report
-        if st.button("📄 Generate Final Report"):
-            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}**")
-            progress_bar.progress(100)
+        # Generate Final Report Automatically
+        progress_bar.progress(99)
+        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}**")
+        progress_bar.progress(100)
+
+    # Save Data to Hugging Face
+    if st.session_state.company_data and st.session_state.ai_strategy:
+        user_data = {
+            "name": name, "email": email, "mobile": mobile, "company_name": company_name, 
+            "industry": company_industry, "input_method": input_method, "company_data": st.session_state.company_data,
+            "ai_strategy": st.session_state.ai_strategy, "ai_integration": st.session_state.ai_integration,
+            "revenue_opportunities": st.session_state.revenue_opportunities, "bni_benefits": st.session_state.bni_benefits
+        }
+        save_data_hf(user_data)
 
     # Footer
     st.markdown("---")
     st.markdown("🚀 **Developed by [Aditya Ghadge](https://www.linkedin.com/in/aditya-ghadge-a82b30240/) for Giant Analytics**")
 
 if __name__ == "__main__":
-    main()
+    main()