import streamlit as st
from langchain_helper import generate_json


st.title("Agent Information Generator")

# Text input box for the user to enter information
text_input = st.text_area("Enter text:", height=200)

if st.button("Generate JSON"):
    if text_input:
        # Generate JSON using OpenAI chat model
        generated_json = generate_json(text_input)

        # Display the generated JSON
        st.json(generated_json)
    else:
        st.warning("Please enter text before generating JSON.")