Spaces:
Sleeping
Sleeping
File size: 3,326 Bytes
e1283cc 343468f 2b2818d 343468f e8b7d05 343468f f54fc8b 343468f e1283cc 343468f e1283cc e83ea81 e1283cc e83ea81 e1283cc b213c2c e83ea81 f54fc8b b213c2c e1283cc e8b7d05 e1283cc 343468f e8b7d05 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
from src.langgraphagenticai.ui.streamlitui.display_result import DisplayResultStreamlit
from src.langgraphagenticai.LLMS.groqllm import GroqLLM
from src.langgraphagenticai.graph.graph_builder import GraphBuilder
from src.langgraphagenticai.ui.streamlitui.loadui import LoadStreamlitUI
import streamlit as st
import json
# MAIN Function START
def load_langgraph_agenticai_app():
"""
Loads and runs the LangGraph AgenticAI application with Streamlit UI.
This function initializes the UI, handles user input, configures the LLM model,
sets up the graph based on the selected use case, and displays the output while
implementing exception handling for robustness.
"""
try:
# Load UI
ui = LoadStreamlitUI()
user_input = ui.load_streamlit_ui()
if not user_input:
st.error("Error: Failed to load user input from the UI.")
return
user_message = ''
# Text input for user message
if st.session_state.IsFetchButtonClicked:
user_message = st.session_state.timeframe
elif st.session_state.IsSDLC or 'current_step' in st.session_state.state:
user_message = st.session_state.state
else :
user_message = st.chat_input("Enter your message:")
if user_message:
try:
# Configure LLM
obj_llm_config = GroqLLM(user_controls_input=user_input)
model = obj_llm_config.get_llm_model()
if not model:
st.error("Error: LLM model could not be initialized.")
return
# Initialize and set up the graph based on use case
usecase = user_input.get('selected_usecase')
if not usecase:
st.error("Error: No use case selected.")
return
graph_builder = GraphBuilder(model)
try:
graph = graph_builder.setup_graph(usecase)
except Exception as e:
st.error(f"Error: Graph setup failed - {e}")
return
# Display output in UI
try:
# Add travel-specific parameters to message
if usecase == "Travel Planner":
travel_params = {
'source' : user_input.get('source'),
'city': user_input.get('destination'),
'start_date': user_input.get('start_date').isoformat(),
'end_date': user_input.get('end_date').isoformat(),
'interests': user_input.get('preferences'),
'user_message' : user_message
}
user_message = travel_params
DisplayResultStreamlit(usecase,graph,user_message).display_result_on_ui()
except Exception as e:
st.error(f"Error: Failed to display results on UI - {e}")
except Exception as e:
st.error(f"Error: LLM configuration failed - {e}")
except Exception as e:
st.error(f"Unexpected error occurred: {e}")
|