Update LegalBot2.py

LegalBot2.py

@@ -1,31 +1,35 @@
 import os
 import streamlit as st
-from langchain_openai import OpenAIEmbeddings
-from langchain_openai.chat_models import ChatOpenAI
-from langchain_community.vectorstores import Chroma
+from langchain.embeddings.openai import OpenAIEmbeddings
+from langchain.vectorstores import Chroma
+from langchain.chat_models import ChatOpenAI
 from langchain.chains import RetrievalQA
-from langchain.prompts import PromptTemplate
+from langchain import PromptTemplate
 from langchain.memory import ConversationBufferMemory
-from langchain.agents import initialize_agent, Tool
+from langchain.agents import initialize_agent, Tool, AgentExecutor
 from langchain.text_splitter import CharacterTextSplitter
 import openai
 
-# Set OpenAI API Key securely
-openai_api_key = os.environ.get("OPENAI_API_KEY")
-if not openai_api_key:
-    raise ValueError("OpenAI API key is not set in environment variables.")
-os.environ["OPENAI_API_KEY"] = openai_api_key
+# Set OpenAI API Key
+OPENAI_API_KEY = "sk-XC6vDUoMgm_qjeT-iAKtJOrPHLZZAupnjO39kh_x9nT3BlbkFJ-tfz3_p02QWfnC77SnWI32IPWLAqtLnX_VMlBpTH4A"
+os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
+openai.api_key = OPENAI_API_KEY
 
 # Document file paths
 file1 = "./data/DIVISION OF ASSETS AFTER DIVORCE.txt"
 file2 = "./data/INHERITANCE.txt"
 
+
+# Function to initialize the OpenAI embeddings and model
 def openai_setting():
     embedding = OpenAIEmbeddings()
-    model_name = "gpt-3.5-turbo"
+    # model_name = "gpt-3.5-turbo"
+    model_name = "gpt-4o-mini"
     llm = ChatOpenAI(model_name=model_name, temperature=0)
     return embedding, llm
 
+
+# Function to split the law content
 def law_content_splitter(path, splitter="CIVIL CODE"):
     with open(path) as f:
         law_content = f.read()
@@ -33,28 +37,61 @@ def law_content_splitter(path, splitter="CIVIL CODE"):
     text_splitter = CharacterTextSplitter()
     return text_splitter.create_documents(law_content_by_article)
 
+
+# Function to determine if input is a greeting
 def is_greeting(input_str):
     greetings = [
-        "hello", "hi", "hey", "greetings", "good morning", "good afternoon", 
-        "good evening", "hi there", "hello there", "hey there", 
-        "whats up", "ciao", "salve", "buongiorno", 
-        "buona sera", "buonasera", "buon pomeriggio", "buonpomeriggio", 
-        "come stai", "comestai", "come va", "comeva", "come sta", "comesta"
+        "hello",
+        "hi",
+        "hey",
+        "greetings",
+        "good morning",
+        "good afternoon",
+        "good evening",
+        "hi there",
+        "hello there",
+        "hey there",
+        "whats up",
+        "ciao",
+        "salve",
+        "buongiorno",
+        "buona sera",
+        "buonasera",
+        "buon pomeriggio",
+        "buonpomeriggio",
+        "come stai",
+        "comestai",
+        "come va",
+        "comeva",
+        "come sta",
+        "comesta",
     ]
     return any(greet in input_str.lower() for greet in greetings)
 
-def chatbot1(question, agent):
+
+# Function to handle chatbot logic
+def chatbot1(question):
     try:
         return agent.run(question)
     except Exception as e:
-        print(f"Error: {e}")
-        return "I'm sorry, I'm having trouble understanding your question."
+        return f"I'm sorry, I'm having trouble understanding your question. Error: {str(e)}"
 
-def chatbot(input_str, agent):
+
+def chatbot(input_str):
     if is_greeting(input_str):
         return "Hello! Ask me your question about Italian Divorce or Inheritance Law?"
     else:
-        return chatbot1(input_str, agent)
+        return chatbot1(input_str)
+
+
+# If you want to disable Greeting in the chatbot, use this code:
+# def chatbot(input_str):
+#     # Directly process every input as a question
+#     response = chatbot1(input_str)
+#     if response == "N/A":
+#         return "I'm sorry, I'm having trouble understanding your question. Could you please rephrase it or provide more context"
+#     else:
+#         return response
 
 # Splitting the content of law documents
 divorce_splitted = law_content_splitter(file1)
@@ -74,12 +111,15 @@ You should capably address queries regarding asset allocation, child custody, sp
 {context}
 
 Question: {question}"""
-DIVORCE_BOT_PROMPT = PromptTemplate(template=divorce_prompt, input_variables=["context", "question"])
+DIVORCE_BOT_PROMPT = PromptTemplate(
+    template=divorce_prompt, input_variables=["context", "question"]
+)
 
+# Define inheritance prompt
 inheritance_prompt = """As a specialist in Italian inheritance law, you should deliver detailed and accurate insights about inheritance regulations in Italy.
 You should always cite the article numbers you reference. 
 When responding to user queries, you should always base your answers on the provided context.
-Always MUST MUST cite the specific article numbers you mention and refrain from speculating.
+Always MUST cite the specific article numbers you mention and refrain from speculating.
 Maintain precision in all your responses.
 If a user's question doesn't align with the legal documents, you should point out that it's beyond your domain of expertise.
 You should elucidate Italian inheritance law comprehensively, touching on topics such as testamentary inheritance, intestate inheritance, and other pertinent subjects.
@@ -91,7 +131,9 @@ You should provide detailed information on tax nuances associated with inheritan
 {context}
 
 Question: {question}"""
-INHERITANCE_BOT_PROMPT = PromptTemplate(template=inheritance_prompt, input_variables=["context", "question"])
+INHERITANCE_BOT_PROMPT = PromptTemplate(
+    template=inheritance_prompt, input_variables=["context", "question"]
+)
 
 # Setup for Chroma databases and RetrievalQA
 chroma_directory = "./docs/chroma"
@@ -123,30 +165,38 @@ tools = [
     Tool(
         name="Divorce Italian law QA System",
         func=divorce.run,
-        description="useful for when you need to answer questions about divorce laws in Italy. Give also the number of article you use for it.",
+        description="Useful for when you need to answer questions about divorce laws in Italy. Also provides the number of the article you use.",
     ),
     Tool(
         name="Inheritance Italian law QA System",
         func=inheritance.run,
-        description="useful for when you need to answer questions about inheritance laws in Italy. Give also the number of article you use for it.",
+        description="Useful for when you need to answer questions about inheritance laws in Italy. Also provides the number of the article you use.",
     ),
 ]
 
 # Initialize conversation memory and ReAct agent
-memory = ConversationBufferMemory(memory_key="chat_history", input_key="input", output_key="output")
-agent = initialize_agent(tools, llm, agent="zero-shot-react-description", memory=memory, verbose=False)
+memory = ConversationBufferMemory(
+    memory_key="chat_history", input_key="input", output_key="output"
+)
+react = initialize_agent(tools, llm, agent="zero-shot-react-description")
+agent = AgentExecutor.from_agent_and_tools(
+    tools=tools, agent=react.agent, memory=memory, verbose=False
+)
+
 
 # Streamlit UI Setup
-def setup_ui(agent):
+def setup_ui():
     st.set_page_config(page_title="Italian Law Chatbot", page_icon="⚖️")
     st.title("🏛️ Legal Chatbot: Divorce and Inheritance Italy Laws ")
 
-    st.write("""
+    st.write(
+        """
     [![HuggingFace Space](https://huggingface.co/datasets/huggingface/badges/resolve/main/open-in-hf-spaces-sm.svg)](https://huggingface.co/spaces/sattari/legal-chat-bot/tree/main)
     [![Github Repository](https://img.shields.io/badge/GitHub%20Repository-gray?logo=github)](https://github.com/pouyasattari/Legal-Chatbot-italy-divorce-inheritance)
     [![SATTARI.org](https://img.shields.io/badge/SATTARI.org-gray?logo=internetexplorer)](https://www.sattari.org)
     ![Visitors](https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fsattari-legal-chat-bot.hf.space&label=Visitors&labelColor=%235d5d5d&countColor=%231e7ebf&style=flat)
-    """)
+    """
+    )
 
     st.info(
         "Check out full tutorial to build this app on Streamlit [📝 blog](https://sattari.org/legal-chatbot-divorce-and-inheritance-italy-laws/)",
@@ -157,7 +207,6 @@ def setup_ui(agent):
         "Check out [Prompt Examples List](https://github.com/pouyasattari/Legal-Chatbot-italy-divorce-inheritance/blob/main/promptExamples.txt) to know how to interact with this ChatBot 🤗 ",
         icon="✅",
     )
-
     if "messages" not in st.session_state:
         st.session_state.messages = [
             {
@@ -171,7 +220,9 @@ def setup_ui(agent):
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
-    if user_input := st.chat_input("Ask a question about Italian Divorce or Inheritance Law:"):
+    if user_input := st.chat_input(
+        "Ask a question about Italian Divorce or Inheritance Law:"
+    ):
         st.session_state.messages.append({"role": "user", "content": user_input})
         with st.chat_message("user"):
             st.markdown(user_input)
@@ -179,11 +230,14 @@ def setup_ui(agent):
         # Generate and display chatbot response
         with st.chat_message("assistant"):
             response_placeholder = st.empty()
-            response = chatbot(user_input, agent)  # Pass the agent to the chatbot function
+            response = chatbot(user_input)  # Your existing chatbot function
             response_placeholder.markdown(response)
 
         # Append the response to the conversation history
         st.session_state.messages.append({"role": "assistant", "content": response})
 
+
+## Made by Pouya /  www.SATTARI.org
+
 if __name__ == "__main__":
-    setup_ui(agent)
+    setup_ui()