Laeeeq/React_Agent_App

1import streamlit as st2from agents.react_agent import agent3from utils.custom_logger import get_logger4from utils.exception import CustomException5 6# Initialize logger7logger = get_logger(__name__)8 9def get_agent_response(query):10    try:11        logger.info(f"Executing query: {query}")12        # Use the agent to get a response for the provided query13        response = agent.chat(query)14        return response15    except CustomException as ce:16        logger.error(f"Custom exception occurred: {ce}")17        return f"Error: {ce}"18    except Exception as e:19        logger.error(f"Unhandled exception: {str(e)}")20        return f"Error: {str(e)}"21 22def main():23    st.title("ReAct LLM Agent")24    25    # Query input26    query = st.text_input("Enter your query:", "")27    28    if query:29        # Get agent response for the input query30        response = get_agent_response(query)31        32        # Display the agent's response33        st.write("Agent's Response:")34        st.write(response)35    else:36        st.write("Please enter a query to begin.")37 38if __name__ == "__main__":39    main()40