import streamlit as st 
st.set_page_config(layout="wide", page_title="LangGraph AI Agent")
st.title("LangGraph AI Agent")
st.markdown("This is a simple interface to interact with the LangGraph AI Agent. The AI Agent is a language model that can generate text based on a prompt. You can use this interface to chat with the AI Agent and see the responses it generates.")

system_prompt= st.text_area('Define you AI Agent:', placeholder='Type your prompt here...')

MODEL_NAMES_GROQ = ['llama3-70b-8192', 'mixtral-8x7b-32768', 'llama-3.3-70b-versatile']

MODEL_NAMES_OPENAI = ['gpt-4o-mini']

provider= st.selectbox('Select the model provider:', ['Groq', 'OpenAI'])

model_name= st.selectbox('Select the model name:', MODEL_NAMES_GROQ if provider=='Groq' else MODEL_NAMES_OPENAI)

allow_search= st.checkbox('Allow search')
user_query= st.text_area('Enter query:', placeholder='Type your query here...', height=150)
API_URL ='http://127.0.0.1:8000/chat'
if st.button('Chat'):
    if user_query.strip():
        import requests 
        response = requests.post(API_URL, json={
            'model_name': model_name,
            'model_provider': provider,
            'system_prompt': system_prompt,
            'messages': [user_query],
            'allow_search': allow_search
        })
        if response.status_code == 200:
            response_data = response.json()
            if 'error' in response_data:
                st.error(response_data['error'])
            st.markdown(f'AI Agent Response:{response_data}')