Update app.py

app.py

@@ -36,21 +36,10 @@ def main():
     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 = ""
+    # Initialize session state for persistent data
+    for key in ["company_data", "industry_trends", "ai_use_cases", "competitor_analysis", "ai_strategy", "ai_integration", "revenue_opportunities"]:
+        if key not in st.session_state:
+            st.session_state[key] = ""
     
     # Collect User Information
     name = st.text_input("Name")
@@ -65,18 +54,22 @@ def main():
     if input_method == "Search by Name":
         if st.button("Find Company Details"):
             st.session_state.company_data = search_company(company_name)
+            st.experimental_rerun()
     elif input_method == "Website URL":
         website_url = st.text_input("Enter Website URL")
         if st.button("Scrape Website"):
             st.session_state.company_data = scrape_website(website_url)
+            st.experimental_rerun()
     elif input_method == "Manual Description":
         company_data_input = st.text_area("Enter Company Description")
         if st.button("Process Description"):
             st.session_state.company_data = process_company_description(company_data_input)
+            st.experimental_rerun()
     elif input_method == "Upload Document":
         uploaded_file = st.file_uploader("Upload PDF or PPT", type=["pdf", "pptx"])
         if uploaded_file is not None:
             st.session_state.company_data = process_uploaded_document(uploaded_file)
+            st.experimental_rerun()
     
     if st.session_state.company_data:
         st.subheader("Extracted Company Information")
@@ -85,6 +78,7 @@ def main():
         industry = st.text_input("Industry Type (e.g., Healthcare, Finance)")
         if st.button("Analyze Industry Trends"):
             st.session_state.industry_trends = get_industry_trends(industry)
+            st.experimental_rerun()
         
         if st.session_state.industry_trends:
             st.subheader("Industry Trends")
@@ -92,6 +86,7 @@ def main():
             
         if st.button("Find AI Use Cases"):
             st.session_state.ai_use_cases = get_ai_use_cases(industry)
+            st.experimental_rerun()
         
         if st.session_state.ai_use_cases:
             st.subheader("AI Use Cases")
@@ -100,6 +95,7 @@ def main():
         competitor = st.text_input("Enter Competitor Name")
         if st.button("Analyze Competitor AI Strategies"):
             st.session_state.competitor_analysis = get_competitor_ai_strategies(competitor)
+            st.experimental_rerun()
         
         if st.session_state.competitor_analysis:
             st.subheader("Competitor AI Strategies")
@@ -109,6 +105,7 @@ def main():
             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)
+            st.experimental_rerun()
         
         if st.session_state.ai_strategy:
             st.subheader("AI Strategy")
@@ -117,6 +114,7 @@ def main():
         if st.button("Suggest AI Integration Plan"):
             st.session_state.ai_integration = suggest_ai_integration(
                 st.session_state.company_data, st.session_state.ai_strategy)
+            st.experimental_rerun()
         
         if st.session_state.ai_integration:
             st.subheader("AI Integration Plan")
@@ -125,6 +123,7 @@ def main():
         if st.button("Identify Revenue Growth Opportunities"):
             st.session_state.revenue_opportunities = identify_revenue_opportunities(
                 st.session_state.company_data, st.session_state.ai_strategy)
+            st.experimental_rerun()
         
         if st.session_state.revenue_opportunities:
             st.subheader("Revenue Growth Opportunities")
@@ -136,19 +135,8 @@ def main():
                 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": 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**")
+    st.markdown("---")
+    st.markdown("**Developed by Aditya Ghadge for Giant Analytics**")
 
 if __name__ == "__main__":
-    main()
+    main()