import streamlit as st
from user_input import get_user_input
from ai_processing import generate_company_profile, calculate_fundraising_score, generate_recommendations
from visualization import create_comparison_chart

def main():
    st.set_page_config(page_title="BandAI - AI-Powered Fundraising Advisor", layout="wide")
    st.title("BandAI - AI-Powered Fundraising Advisor")

    # Get user input
    user_data = get_user_input()

    if st.button("Analyze"):
        with st.spinner("Analyzing your startup..."):
            # Generate company profile
            company_profile = generate_company_profile(user_data)
            st.subheader("AI-Generated Company Profile")
            st.write(company_profile)

            # Calculate fundraising score
            fundraising_score = calculate_fundraising_score(user_data)
            st.subheader("Fundraising Probability Score")
            st.progress(fundraising_score / 100)
            st.write(f"{fundraising_score}%")

            # Generate recommendations
            recommendations = generate_recommendations(user_data)
            st.subheader("Suggestions for Improving Fundraising Success")
            for i, rec in enumerate(recommendations, 1):
                st.write(f"{i}. {rec}")

            # Create comparison visualization
            comparison_chart = create_comparison_chart(user_data)
            st.subheader("Company Comparison")
            st.plotly_chart(comparison_chart)

if __name__ == "__main__":
    main()