Update app.py

app.py

@@ -1,5 +1,41 @@
 import os
 import streamlit as st
+from typing_extensions import TypedDict, List
+from IPython.display import Image, display
+from langchain_core.pydantic_v1 import BaseModel, Field
+from langchain.schema import Document
+from langgraph.graph import START, END, StateGraph
+from langchain.prompts import PromptTemplate
+import uuid
+from langchain_exa import ExaSearchRetriever, TextContentsOptions
+from langchain_groq import ChatGroq
+from langchain_community.utilities import GoogleSerperAPIWrapper
+from langchain_chroma import Chroma
+from langchain_community.document_loaders import NewsURLLoader
+from langchain_community.retrievers.wikipedia import WikipediaRetriever
+from sentence_transformers import SentenceTransformer
+from langchain.vectorstores import Chroma
+from langchain_community.document_loaders import UnstructuredURLLoader, NewsURLLoader
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain.text_splitter import RecursiveCharacterTextSplitter
+from langchain_community.document_loaders import WebBaseLoader
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.output_parsers import JsonOutputParser
+from langchain_community.vectorstores.utils import filter_complex_metadata
+from langchain.schema import Document
+from langgraph.graph import START, END, StateGraph
+from functions import *
+
+
+
+os.environ["LANGCHAIN_TRACING_V2"] = "true"
+os.environ["LANGCHAIN_ENDPOINT"] = "https://api.langchain.plus"
+os.environ["LANGCHAIN_API_KEY"] = lang_api_key
+os.environ["LANGCHAIN_PROJECT"] = "Lithuanian_Law_RAG_QA"
+os.environ["GROQ_API_KEY"] = groq_api_key
+os.environ["SERPER_API_KEY"] = SERPER_API_KEY
+
+
 
 
 
@@ -92,7 +128,7 @@ def main():
 
     
     if user_question := st.text_input("Ask a question about your documents:"):
-    handle_userinput(user_question,retriever,rag_chain)            
+    handle_userinput(user_question, custom_graph)