Upload 6 files

Functions.py

@@ -0,0 +1,133 @@
+from dotenv import load_dotenv
+import os
+import uuid
+from PyPDF2 import PdfReader
+from docx import Document
+from docx.text.paragraph import Paragraph
+from docx.table import Table
+from langchain.text_splitter import CharacterTextSplitter
+from langchain_community.vectorstores import Chroma
+from langchain_community.embeddings import OpenAIEmbeddings
+import streamlit as st
+from textwrap import dedent
+from Prompts_and_Chains import LLMChains
+
+
+def extract_text_from_file(file):
+    text = file.read().decode("utf-8")
+    return text
+
+
+def process_paragraph(paragraph):
+    # Process the content of the paragraph as needed
+    return paragraph.text
+
+
+def process_table(table):
+    # Process the content of the table as needed
+    text = ""
+    for row in table.rows:
+        for cell in row.cells:
+            text += cell.text
+
+    return text
+
+
+def read_docx(file_path):
+    doc = Document(file_path)
+    data = []
+
+    for element in doc.iter_inner_content():
+        if isinstance(element, Paragraph):
+            data.append(process_paragraph(element))
+        if isinstance(element, Table):
+            data.append(process_table(element))
+
+    return "\n".join(data)
+
+
+def get_pdf_text(pdf):
+    """This function extracts the text from the PDF file"""
+    text = []
+    pdf_reader = PdfReader(pdf)
+    for page in pdf_reader.pages:
+        text.append(page.extract_text())
+
+    return "\n".join(text)
+
+
+class RFPProcessor:
+    def __init__(self):
+        load_dotenv()
+        self.openai_api_key = os.getenv("OPENAI_API_KEY")
+        self.chains_obj = LLMChains()
+
+    def process_case_data(self, case_name, files):
+        if case_name and files:
+            # Generate a unique identifier for the case data set
+            case_id = str(uuid.uuid4())
+            extracted_data = []
+            all_texts = []
+            for file in files:
+                file_text = []
+                if file.name.endswith(".docx"):
+                    file_text = read_docx(file)
+                elif file.name.endswith(".pdf"):
+                    file_text = get_pdf_text(file)
+                else:
+                    file_text = extract_text_from_file(file)
+
+                text_splitter = CharacterTextSplitter(
+                    separator="\n", chunk_size=1000, chunk_overlap=150, length_function=len
+                )
+                texts = text_splitter.split_text(" ".join(file_text))
+
+                all_texts.extend(texts)
+                extracted_data.append(" ".join(file_text))
+
+            project_dir = os.path.dirname(os.path.abspath(__file__))
+            vectorstore = Chroma(
+                persist_directory=os.path.join(
+                    project_dir, "vector_stores", case_name),
+                embedding_function=OpenAIEmbeddings(
+                    openai_api_key=self.openai_api_key),
+            )
+            vectorstore.add_texts(all_texts)
+
+            st.session_state[case_id] = {
+                "vectorstore": vectorstore,
+                "extracted_data": extracted_data,
+            }
+            all_text = " ".join(extracted_data)
+            st.session_state["case_summary"] = self.chains_obj.case_summary_chain.run(
+                {
+                    "case_name": case_name,
+                    "case_info": dedent(all_text),
+                }
+            )
+            st.session_state["is_data_processed"] = True
+            st.session_state["case_name"] = case_name
+            st.session_state["case_details"] = dedent(all_text)
+            st.session_state["current_case_id"] = case_id
+            st.success("Data processed successfully")
+
+    def genrate_legal_bot_result(self):
+        if len(st.session_state["bot_input"]) > 0:
+            case_id = st.session_state.get("current_case_id")
+            if case_id:
+                vector_store = st.session_state[case_id]["vectorstore"]
+                query = st.session_state["bot_input"]
+                results = vector_store.similarity_search(query, 3)
+                # get the text from the results
+                source_knowledge = "\n".join([x.page_content for x in results])
+                inputs = {
+                    "case_summary":st.session_state["case_summary"],
+                    "context": source_knowledge,
+                    "input": query,
+                }
+                output = self.chains_obj.legal_case_bot_chain.run(inputs)
+                st.session_state.past_bot_results.append(st.session_state["bot_input"])
+                st.session_state.generated_bot_results.append(output)
+                st.session_state["bot_input"] = ""
+            else:
+                st.warning(f"No vector store found for the current case ID")

Prompts_and_Chains.py

@@ -0,0 +1,63 @@
+# ******* THIS FILE CONTAINS ALL THE PROMPTS & CHAINS USED IN Functions.py ***********
+from Templates import *
+from langchain import PromptTemplate
+from langchain.chains import LLMChain
+from langchain.llms import OpenAI
+from dotenv import load_dotenv
+import os
+import streamlit as st
+
+
+class PromptTemplates:
+    def __init__(self):
+
+        self.legal_adviser_bot_prompt = PromptTemplate(
+            input_variables=["chat_history","input",], template=legal_adviser_template
+        )
+
+        self.case_summary_prompt = PromptTemplate(
+            input_variables=["case_name", "case_info"], template=case_summary_template
+        )
+
+        self.legal_case_bot_prompt = PromptTemplate(
+            input_variables=["case_summary", "context","input"], template=legal_case_bot_template
+        )
+
+        self.lawyer_recommendations_prompt = PromptTemplate(
+            input_variables=["user_inputs", "matching_lawyers", "additional_info"], template=lawyer_recommendation_template
+        )
+
+
+class LLMChains:
+    def __init__(self):
+        load_dotenv()
+        openai_api_key = os.getenv("OPENAI_API_KEY")
+        obj = PromptTemplates()
+        model_name = st.session_state["selected_model"]
+
+        # generate summary chain
+        self.legal_adviser_bot_chain = LLMChain(
+            llm=OpenAI(model_name='gpt-3.5-turbo-16k', temperature=0.7),
+            prompt=obj.legal_adviser_bot_prompt,
+            verbose="true",
+        )
+
+        # genrate bot conversastion
+        self.case_summary_chain = LLMChain(
+            llm=OpenAI(model_name=model_name, temperature=0.7),
+            prompt=obj.case_summary_prompt,
+            verbose="true",
+        )
+
+        # genrate bot conversastion
+        self.legal_case_bot_chain = LLMChain(
+            llm=OpenAI(model_name=model_name, temperature=0.7),
+            prompt=obj.legal_case_bot_prompt,
+            verbose="true",
+        )
+
+        self.lawyer_recommendations_chain = LLMChain(
+            llm=OpenAI(model_name="gpt-3.5-turbo-16k", temperature=0.7),
+            prompt=obj.lawyer_recommendations_prompt,
+            verbose="true",
+        )

Templates.py

@@ -0,0 +1,128 @@
+# ******* THIS FILE CONTAINS ALL THE TEMPLATES USED IN APP.PY ***********
+
+
+case_summary_template = """
+You are an experienced legal expert with a comprehensive understanding of various areas of law. Your background includes:
+
+- Over 20 years of experience in legal practice and analysis
+- Expertise in multiple fields including civil law, criminal law, corporate law, and constitutional law
+- A track record of successfully summarizing complex legal cases for both legal professionals and laypeople
+- An ability to identify key legal issues and present them clearly and concisely
+
+Your task is to summarize the following case information. In your summary, please include:
+
+1. Key parties involved
+2. Main legal issues at stake
+3. Important facts that are crucial to the case
+4. Procedural history (how the case has progressed through the legal system)
+5. Outcome or current status of the case
+
+Focus on the most crucial information and present it in a clear, organized manner. Use your expertise to highlight any particularly significant or unusual aspects of the case.
+
+Case Name :{case_name}
+Case infromation:{case_info}
+
+"""
+
+legal_case_bot_template = """ 
+You are a highly knowledgeable legal expert with extensive experience in case analysis and legal consultation. Your role is to assist with questions related to the following case:
+
+Case Summary:
+{case_summary}
+
+Additional Context:
+{context}
+
+Drawing upon your expertise and the information provided, please answer the following question accurately and concisely. If the answer cannot be determined from the available information, clearly state this and explain what additional information would be needed to provide a complete answer.
+User Question: {input}
+Assistant:"""
+
+
+# legal_adviser_template = """
+# You are LegalAssist, an AI chatbot specializing in Indian legal information. Your purpose is to provide general information about Indian law and legal matters. Always begin your interaction with a brief introduction explaining your role and limitations.
+
+# Your capabilities:
+# 1. Answer questions about general legal concepts and definitions in Indian law
+# 2. Provide basic information on different areas of Indian law
+# 3. Explain legal procedures and processes in the Indian legal system
+# 4. Help users understand Indian legal documents
+# 5. Offer information on legal rights and responsibilities under Indian law
+
+# Your limitations:
+# 1. You cannot provide personalized legal advice
+# 2. You are not a licensed attorney in India
+# 3. Your responses should not be considered as legal counsel
+
+# Guidelines:
+# 1. Always maintain a professional and neutral tone
+# 2. If a question is unclear, ask for clarification
+# 3. If a query is outside the scope of Indian legal matters, politely explain that you can only assist with Indian law-related questions
+# 4. Remind users not to share personal or confidential information
+# 5. If a question requires legal expertise beyond general knowledge of Indian law, advise the user to consult with a qualified Indian lawyer
+
+# For off-topic questions, respond with:
+# "I apologize, but I'm not able to assist with that question as it's outside my area of expertise. I'm designed specifically to answer questions related to Indian law and legal matters. Is there an Indian legal topic you'd like information about?"
+
+# Remember, your primary function is to provide helpful, accurate, and general information about Indian law while clearly communicating your limitations as an AI assistant.
+
+# User Query: {input}
+
+# Please respond to the above user query following all the guidelines provided, focusing specifically on Indian law.
+# """
+
+legal_adviser_template = """
+You are LegalAssist, an AI chatbot specializing in legal information and you personality is legal assiatant. Your purpose is to provide general information about law and legal matters. Always begin your interaction with a brief introduction explaining your role and limitations.
+
+Your capabilities:
+1. Answer questions about general legal concepts and definitions
+2. Provide basic information on different areas of law
+3. Explain legal procedures and processes
+4. Help users understand legal documents
+5. Offer information on legal rights and responsibilities
+6. Provide information about different types of lawyers and their roles
+7. Explain sections of the Indian Penal Code (IPC)
+8. Provide details about various acts and laws in India
+9. Explain articles of the Indian Constitution
+10. Provide information on amendments to Indian laws
+11. you have the capability to remember previous questions
+
+Your limitations:
+1. You cannot provide personalized legal advice
+2. You are not a licensed attorney
+3. Your responses should not be considered as legal counsel
+
+Guidelines:
+1. Always maintain a professional and neutral tone
+2. If a question is unclear, ask for clarification
+3. If a query is outside the scope of your personality apologize for that
+4. Remind users not to share personal or confidential information
+5. If a question requires legal expertise beyond general knowledge, advise the user to consult with a qualified lawyer
+6. Ensure your chatbot can handle follow-up questions and provide clarifications and use privous chat given blow.
+7. Ensure the chatbot's ability to handle common variations of questions.
+
+
+For off-topic questions, respond with:
+"I apologize, but I'm not able to assist with that question as it's outside my area of expertise. I'm designed specifically to answer questions related to law and legal matters. Is there a legal topic you'd like information about?"
+
+Remember, your primary function is to provide helpful, accurate, and general legal information while clearly communicating your limitations as an AI assistant.
+
+{chat_history}
+
+use this chat_history to answer the follow up questions and previous questions
+
+Human: {input}
+Assistant:
+"""
+
+
+lawyer_recommendation_template = """
+Based on the user's inputs and the matching lawyers, provide a recommendation for the best lawyer(s).
+Consider the user's specific needs mentioned in the additional information, and pay special attention to the lawyers' descriptions and expertise.
+
+User inputs: {user_inputs}
+Matching lawyers: {matching_lawyers}
+Additional information: {additional_info}
+
+Provide a detailed recommendation of lawyers witht lawyer information , explaining why you think this lawyer or these lawyers would be the best fit for the user's needs. Consider their specialties, experience, and how their expertise aligns with the user's case details.
+
+"""

app.py

@@ -0,0 +1,122 @@
+from dotenv import load_dotenv
+import os
+import streamlit as st
+from streamlit_option_menu import option_menu
+from Functions import RFPProcessor
+
+from utills import ( on_click_toggle)
+
+ 
+# Initialize session states for rfp data (data processing)
+if "vectorstore" not in st.session_state:
+    st.session_state["vectorstore"] = None
+if "pdf_details" not in st.session_state:
+    st.session_state["pdf_details"] = ""
+if "is_data_processed" not in st.session_state:
+    st.session_state["is_data_processed"] = False
+    
+
+# Initialize session states for summary  bot
+if "generated_bot_results" not in st.session_state:
+    st.session_state["generated_bot_results"] = []
+if "past_bot_results" not in st.session_state:
+    st.session_state["past_bot_results"] = []
+if "bot_input" not in st.session_state:
+    st.session_state["bot_input"] = "" 
+
+
+# Initialize session states for case
+if "case_summary" not in st.session_state:
+    st.session_state["case_summary"] = ""
+    
+if "selected_model" not in st.session_state:
+    st.session_state["selected_model"] = "gpt-3.5-turbo-16k"
+    
+   
+def main():
+    st.set_page_config(page_title="Justice League",
+                       page_icon="⚖️", layout="wide")
+    top_left_container = st.sidebar.container()
+    model_name = "GPT-4o" if st.session_state["selected_model"] == "gpt-4o" else "GPT-3.5-turbo"
+    st.markdown(
+            """
+            <style>
+             .st-ae.st-af.st-ag.st-ah.st-ai.st-aj.st-ak.st-al.st-am {
+                position: absolute !important;
+                top: -80px !important;
+                z-index: 9999;
+            }
+            </style>
+            """,
+            unsafe_allow_html=True
+        )
+    with top_left_container:
+        st.markdown("<div style='margin-bottom: -10px; font-size: 15px; position: absolute; z-index: 1; top: -91px; left: 3px'>Toggle for Model Selection</div>", unsafe_allow_html=True)
+        st.markdown("<div style='margin-bottom: -10px; font-size: 15px; position: absolute; z-index: 1; top: -87px; left: 3px'>(Default : GPT-3.5)</div>", unsafe_allow_html=True)
+        st.toggle(model_name, value=st.session_state["selected_model"] == "gpt-4o", key="model_toggle", on_change=on_click_toggle, args=None, disabled=False, label_visibility="visible") 
+    
+    function = RFPProcessor()
+
+    with st.sidebar:
+        menu_choice = option_menu(
+            menu_title="Legal Assistant",
+            options=[
+                "Home",
+                "Legal Bot",
+            ],
+            icons=["house", "list-task"],
+        )
+                    
+             
+    if menu_choice == "Home":
+        with st.form("my_form"):
+            case_name = st.text_input(
+                "Case Name",
+                key="Case Name",
+                type="default",
+                placeholder="Case Name",
+            )
+            files = st.file_uploader(
+                "Document", type=["pdf", "txt", "docx"], accept_multiple_files=True
+            )
+
+            submitted = st.form_submit_button("Process Data")
+
+            if submitted:
+                if case_name and files:
+                    function.process_case_data(case_name, files)
+                else:
+                    st.warning(
+                        "project_name and file are required to create create stories",
+                        icon="⚠️",
+                    )
+        if st.session_state["is_data_processed"] == True:
+            st.title("Summary")
+            with st.container():
+                st.markdown(st.session_state["case_summary"])
+
+    if menu_choice == "Legal Bot":
+        if st.session_state["is_data_processed"] == True:
+            st.title(" Legal Bot ")
+            st.subheader(" Powered by Coffeebeans")
+            st.text_input(
+                "You: ",
+                st.session_state["bot_input"],
+                key="bot_input",
+                placeholder="Your AI Case assistant here! Ask me Queries related to Case",
+                on_change=function.genrate_legal_bot_result(),
+                label_visibility="hidden",
+            )
+            with st.container():
+                for i in range(len(st.session_state["generated_bot_results"]) - 1, -1, -1):
+                    st.success(st.session_state["generated_bot_results"][i], icon="🤖")
+                    st.info(st.session_state["past_bot_results"][i], icon="🧐")
+        else:
+            st.warning("Plesase Process Case Details to access this logal case bot", icon="⚠️")
+
+    
+if __name__ == "__main__":
+    
+    main()
+    
+

requirements.txt

@@ -0,0 +1,14 @@
+streamlit==1.31.1
+streamlit-option-menu==0.3.6
+streamlit-chat==0.1.1
+streamlit-lottie==0.0.5
+langchain==0.1.11
+langchain-community==0.0.27
+langchain-core==0.1.30
+langchain-text-splitters==0.0.1
+chromadb==0.4.24
+openai==0.28.0
+python-dotenv==1.0.1
+python-docx==1.1.0
+PyPDF2==3.0.1
+tiktoken==0.7.0

utills.py

@@ -0,0 +1,89 @@
+import streamlit as st
+
+lawyer_db = [
+    {"id": 1, "name": "Rajesh Kumar", "specialty": "Criminal", "experience": "11-20 years", "cost_range": "High", "location": "Jabalpur", "rating": 4.8, "description": "Specializes in IPC cases, has successfully defended in multiple murder trials."},
+    {"id": 2, "name": "Priya Sharma", "specialty": "Family", "experience": "6-10 years", "cost_range": "Medium", "location": "Bhopal", "rating": 4.5, "description": "Expert in divorce cases and child custody battles under Hindu Marriage Act."},
+    {"id": 3, "name": "Amit Patel", "specialty": "Corporate", "experience": "20+ years", "cost_range": "Very High", "location": "Indore", "rating": 4.9, "description": "Handled major mergers and acquisitions, expert in Company Law and SEBI regulations."},
+    {"id": 4, "name": "Neha Gupta", "specialty": "Immigration", "experience": "0-5 years", "cost_range": "Low", "location": "Gwalior", "rating": 4.2, "description": "Specializes in student visas and work permits for IT professionals."},
+    {"id": 5, "name": "Vikram Singh", "specialty": "Criminal", "experience": "11-20 years", "cost_range": "Medium", "location": "Bhopal", "rating": 4.7, "description": "Former public prosecutor, expert in cybercrime cases under IT Act."},
+    {"id": 6, "name": "Anita Desai", "specialty": "Family", "experience": "20+ years", "cost_range": "High", "location": "Indore", "rating": 4.6, "description": "Specializes in inter-faith marriages and Special Marriage Act cases."},
+    {"id": 7, "name": "Sanjay Mehta", "specialty": "Corporate", "experience": "11-20 years", "cost_range": "High", "location": "Jabalpur", "rating": 4.5, "description": "Expert in Intellectual Property Rights and patent law."},
+    {"id": 8, "name": "Kavita Reddy", "specialty": "Immigration", "experience": "6-10 years", "cost_range": "Medium", "location": "Bhopal", "rating": 4.3, "description": "Specializes in NRI property disputes and OCI card issues."},
+    {"id": 9, "name": "Arjun Malhotra", "specialty": "Criminal", "experience": "20+ years", "cost_range": "Very High", "location": "Indore", "rating": 4.9, "description": "Renowned for handling high-profile corruption cases under Prevention of Corruption Act."},
+    {"id": 10, "name": "Deepa Joshi", "specialty": "Family", "experience": "0-5 years", "cost_range": "Low", "location": "Gwalior", "rating": 4.1, "description": "Focuses on domestic violence cases under Protection of Women from Domestic Violence Act."},
+    {"id": 11, "name": "Rahul Verma", "specialty": "Corporate", "experience": "6-10 years", "cost_range": "Medium", "location": "Jabalpur", "rating": 4.4, "description": "Specializes in startup law and venture capital regulations."},
+    {"id": 12, "name": "Sunita Agarwal", "specialty": "Immigration", "experience": "11-20 years", "cost_range": "High", "location": "Bhopal", "rating": 4.7, "description": "Expert in handling complex deportation cases and appeals."},
+    {"id": 13, "name": "Kiran Bedi", "specialty": "Criminal", "experience": "20+ years", "cost_range": "Very High", "location": "Indore", "rating": 4.8, "description": "Former IPS officer, specializes in police brutality and human rights violation cases."},
+    {"id": 14, "name": "Vivek Chauhan", "specialty": "Family", "experience": "11-20 years", "cost_range": "High", "location": "Gwalior", "rating": 4.6, "description": "Expert in adoption cases and surrogacy law in India."},
+    {"id": 15, "name": "Meera Saxena", "specialty": "Corporate", "experience": "20+ years", "cost_range": "Very High", "location": "Jabalpur", "rating": 4.9, "description": "Specializes in banking law and financial regulations, handled major NPAs cases."},
+    {"id": 16, "name": "Alok Nath", "specialty": "Immigration", "experience": "6-10 years", "cost_range": "Medium", "location": "Bhopal", "rating": 4.3, "description": "Expert in handling asylum cases and refugee status applications."},
+    {"id": 17, "name": "Pooja Bhatt", "specialty": "Criminal", "experience": "0-5 years", "cost_range": "Low", "location": "Indore", "rating": 4.2, "description": "Specializes in juvenile justice cases under Juvenile Justice Act."},
+    {"id": 18, "name": "Rohit Sharma", "specialty": "Family", "experience": "11-20 years", "cost_range": "High", "location": "Gwalior", "rating": 4.7, "description": "Expert in handling LGBTQ+ rights cases and same-sex partnership issues."},
+    {"id": 19, "name": "Nandini Roy", "specialty": "Corporate", "experience": "6-10 years", "cost_range": "Medium", "location": "Jabalpur", "rating": 4.5, "description": "Specializes in environmental law and corporate social responsibility cases."},
+    {"id": 20, "name": "Ajay Devgan", "specialty": "Immigration", "experience": "20+ years", "cost_range": "Very High", "location": "Bhopal", "rating": 4.8, "description": "Expert in handling complex citizenship cases and constitutional challenges."},
+    {"id": 21, "name": "Ravi Shankar", "specialty": "Criminal", "experience": "6-10 years", "cost_range": "Medium", "location": "Jabalpur", "rating": 4.4, "description": "Specializes in white-collar crimes and economic offenses."},
+    {"id": 22, "name": "Anjali Kapoor", "specialty": "Family", "experience": "20+ years", "cost_range": "Very High", "location": "Bhopal", "rating": 4.8, "description": "Expert in high-net-worth divorce cases and property disputes."},
+    {"id": 23, "name": "Suresh Patel", "specialty": "Corporate", "experience": "0-5 years", "cost_range": "Low", "location": "Indore", "rating": 4.0, "description": "Focuses on startup incorporation and compliance issues."},
+    {"id": 24, "name": "Preeti Singhania", "specialty": "Immigration", "experience": "11-20 years", "cost_range": "High", "location": "Gwalior", "rating": 4.6, "description": "Specializes in business immigration and investor visas."},
+    {"id": 25, "name": "Arun Jaitley", "specialty": "Criminal", "experience": "20+ years", "cost_range": "Very High", "location": "Jabalpur", "rating": 4.9, "description": "Renowned for handling high-profile criminal cases and constitutional matters."},
+    {"id": 26, "name": "Shalini Mishra", "specialty": "Family", "experience": "6-10 years", "cost_range": "Medium", "location": "Indore", "rating": 4.3, "description": "Specializes in child rights and juvenile law."},
+    {"id": 27, "name": "Vijay Mallya", "specialty": "Corporate", "experience": "11-20 years", "cost_range": "High", "location": "Bhopal", "rating": 4.7, "description": "Expert in corporate restructuring and insolvency cases."},
+    {"id": 28, "name": "Aarti Chabria", "specialty": "Immigration", "experience": "0-5 years", "cost_range": "Low", "location": "Jabalpur", "rating": 4.1, "description": "Focuses on family-based immigration and naturalization cases."},
+    {"id": 29, "name": "Ranveer Kapoor", "specialty": "Criminal", "experience": "6-10 years", "cost_range": "Medium", "location": "Gwalior", "rating": 4.4, "description": "Specializes in drug-related offenses and NDPS Act cases."},
+    {"id": 30, "name": "Sneha Agarwal", "specialty": "Family", "experience": "11-20 years", "cost_range": "High", "location": "Bhopal", "rating": 4.6, "description": "Expert in handling complex custody battles and international family law cases."},
+    {"id": 31, "name": "Raj Malhotra", "specialty": "Corporate", "experience": "20+ years", "cost_range": "Very High", "location": "Indore", "rating": 4.9, "description": "Specializes in cross-border transactions and international trade law."},
+    {"id": 32, "name": "Zoya Khan", "specialty": "Immigration", "experience": "6-10 years", "cost_range": "Medium", "location": "Jabalpur", "rating": 4.3, "description": "Expert in handling visa appeals and deportation defense cases."},
+    {"id": 33, "name": "Aryan Khanna", "specialty": "Criminal", "experience": "0-5 years", "cost_range": "Low", "location": "Bhopal", "rating": 4.0, "description": "Focuses on petty crimes and first-time offender cases."},
+    {"id": 34, "name": "Ishaan Sharma", "specialty": "Family", "experience": "20+ years", "cost_range": "Very High", "location": "Gwalior", "rating": 4.8, "description": "Renowned for handling high-profile divorce cases involving public figures."},
+    {"id": 35, "name": "Aditi Rao", "specialty": "Corporate", "experience": "11-20 years", "cost_range": "High", "location": "Indore", "rating": 4.7, "description": "Specializes in corporate governance and compliance issues."},
+    {"id": 36, "name": "Kabir Singh", "specialty": "Immigration", "experience": "20+ years", "cost_range": "Very High", "location": "Jabalpur", "rating": 4.9, "description": "Expert in complex immigration litigation and federal court appeals."},
+    {"id": 37, "name": "Neha Dhupia", "specialty": "Criminal", "experience": "11-20 years", "cost_range": "High", "location": "Bhopal", "rating": 4.6, "description": "Specializes in cyber law and online fraud cases."},
+    {"id": 38, "name": "Rajat Kapoor", "specialty": "Family", "experience": "0-5 years", "cost_range": "Low", "location": "Indore", "rating": 4.1, "description": "Focuses on family mediation and collaborative divorce processes."},
+    {"id": 39, "name": "Dia Mirza", "specialty": "Corporate", "experience": "6-10 years", "cost_range": "Medium", "location": "Gwalior", "rating": 4.4, "description": "Specializes in entertainment law and intellectual property rights in media."},
+    {"id": 40, "name": "Farhan Akhtar", "specialty": "Immigration", "experience": "11-20 years", "cost_range": "High", "location": "Bhopal", "rating": 4.7, "description": "Expert in handling employment-based immigration and labor certification cases."}
+]
+
+
+def search_lawyers(category, cost_range, experience, location):
+    print(f"Searching for - Category: {category}, Cost Range: {cost_range}, Experience: {experience}, Location: {location}")
+    
+    def find_matches(specialty_match, cost_match, experience_match, location_match):
+        return [
+            lawyer for lawyer in lawyer_db
+            if (specialty_match == "any" or lawyer["specialty"] == category)
+            and (cost_match == "any" or lawyer["cost_range"] == cost_range)
+            and (experience_match == "any" or lawyer["experience"] == experience)
+            and (location_match == "any" or lawyer["location"] == location)
+        ]
+
+    # Try exact match
+    matching_lawyers = find_matches(category, cost_range, experience, location)
+    
+    # Relax location
+    if not matching_lawyers:
+        matching_lawyers = find_matches(category, cost_range, experience, "any")
+        print("Relaxed location criteria")
+    
+    # Relax cost range
+    if not matching_lawyers:
+        matching_lawyers = find_matches(category, "any", experience, "any")
+        print("Relaxed cost range criteria")
+    
+    # Relax experience
+    if not matching_lawyers:
+        matching_lawyers = find_matches(category, "any", "any", "any")
+        print("Relaxed experience criteria")
+    
+    # Relax specialty (return all lawyers if no matches found)
+    if not matching_lawyers:
+        matching_lawyers = find_matches("any", "any", "any", "any")
+        print("Relaxed all criteria")
+
+    print(f"Total matches found: {len(matching_lawyers)}")
+    
+    return matching_lawyers
+
+def on_click_toggle(): 
+    st.session_state.update({
+        "selected_model": "gpt-4o" if st.session_state["selected_model"] == "gpt-3.5-turbo-16k" else "gpt-3.5-turbo-16k"
+    })