Update app.py

app.py

@@ -26,7 +26,7 @@ from langchain.schema import Document
 from langgraph.graph import START, END, StateGraph
 from langchain_community.document_loaders.directory import DirectoryLoader
 from langchain.document_loaders import TextLoader
-from functions import *
+from functions import handle_userinput, create_retriever_from_chroma, retrieval_grader_grader,create_toxicity_checker
 
 
 lang_api_key = os.getenv("lang_api_key")
@@ -119,7 +119,7 @@ def main():
     # Define the nodes
     workflow.add_node("ask_question", lambda state: ask_question(state, retriever))
     workflow.add_node("retrieve", lambda state: retrieve(state, retriever))  
-    workflow.add_node("grade_documents", lambda state: grade_documents(state, retrieval_grader() ))  # grade documents
+    workflow.add_node("grade_documents", lambda state: grade_documents(state, retrieval_grader_grader() ))  # grade documents
     workflow.add_node("generate", generate)  # generatae
     workflow.add_node("web_search", web_search)  # web search
     workflow.add_node("transform_query", transform_query)
@@ -129,7 +129,7 @@ def main():
     workflow.set_entry_point("ask_question")
     workflow.add_conditional_edges(
         "ask_question",
-        lambda state: grade_question_toxicity(state, toxicity_grader()),
+        lambda state: grade_question_toxicity(state, create_toxicity_checker()),
     
         {
         "good": "retrieve",
@@ -151,7 +151,7 @@ def main():
     workflow.add_edge("web_search", "generate")
     workflow.add_conditional_edges(
         "generate",
-        lambda state: grade_generation_v_documents_and_question(state, hallucination_grader(), answer_grader()),
+        lambda state: grade_generation_v_documents_and_question(state, create_hallucination_checker(), create_helpfulness_checker()),
         {
             "not supported": "generate",
             "useful": END,