Update app.py

app.py

@@ -313,9 +313,10 @@ builder.set_entry_point("LLM_Agent")
 
 # Define Conditional Function
 def check_if_answered(state: State) -> str:   
-    if state.get("questions") and not state.get("answers"):
+    if "answers" in state and state['answers'] and any(state['answers']):
+        return "answered"
+    else:
         return "not_answered"
-    return "answered"
 
 # Define Flow
 builder.add_edge("LLM_Agent", "Generate_Questions")
@@ -329,14 +330,13 @@ builder.add_conditional_edges(
     }
 )
 
-
 builder.add_edge("Wait_For_Answers", "Generate_Questions")
 builder.add_edge("Handle_Answers", "Generate_Code")
 builder.add_edge("Generate_Code", "Code_Explainer")
 builder.add_edge("Code_Explainer", END)
 
 # Optionally: define what happens after waiting (if it's a loop)
-builder.add_edge("Wait_For_Answers", "Generate_Questions")  # retry loop
+# builder.add_edge("Wait_For_Answers", "Generate_Questions")  # retry loop
 
 graph = builder.compile(checkpointer=memory) 
 # Streamlit UI setup
@@ -406,7 +406,7 @@ with st.container():
         if submitted and user_input:
             st.session_state.chat_history.append(HumanMessage(content=user_input))
             
-            config = {"configurable": {"thread_id": st.session_state.thread_id}}
+            config = {"configurable": {"thread_id": st.session_state.thread_id},"recursion_limit" : 50}
             state_input = {
                 "messages": st.session_state.chat_history,
                 "input": user_input,