Band_AI / user_input.py
Ashar086's picture
Create user_input.py
88f104e verified
import streamlit as st
def get_user_input():
st.subheader("Company Details")
project_description = st.text_area("Project and Mission Description", height=100)
industry_options = [
"Technology", "Healthcare", "Finance", "Education", "E-commerce",
"Artificial Intelligence", "Blockchain", "Sustainability", "Other"
]
industry = st.selectbox("Industry", industry_options)
market = st.text_input("Target Market")
location = st.text_input("Company Location")
st.subheader("Founders and Team Information")
num_founders = st.number_input("Number of Founders", min_value=1, max_value=10, value=1)
founders_info = []
for i in range(num_founders):
st.write(f"Founder {i+1}")
name = st.text_input(f"Name", key=f"founder_name_{i}")
background = st.text_area(f"Background and Key Strengths", height=50, key=f"founder_background_{i}")
founders_info.append({"name": name, "background": background})
return {
"project_description": project_description,
"industry": industry,
"market": market,
"location": location,
"founders_info": founders_info
}