import streamlit as st
import langchain_helper

st.title("Organization Name Generator!")

domain = st.sidebar.selectbox("Pick a domain you are interested in: ", ("Data Science", "Artificial Intelligence", "Web Development", "Rocket Science", "Astronomy", "Aviation"))

if domain:
    response = langchain_helper.generate_data(domain)
    st.header(response['restaurant_name'].strip())
    menu_items = response['menu_items'].strip().split(",")
    st.write("**Menu Items**")
    for item in menu_items:
        st.write("-", item)