Update app.py

app.py

@@ -291,8 +291,7 @@ two senior most Judges of the Supreme Court, the Union Minister of
 Law & Justice, two eminent persons to be nominated by a committee
 of the Prime Minister of India, the Chief Justice of India, and the
 Leader of the Opposition in the House of the People, or if there is no
-Leader of the Opposition, then the Leader of the single largest
-Opposition Party in the House of the People. Secretary (Justice) will
+Leader of the Opposition, then the Leader of the single largest Opposition Party in the House of the People. Secretary (Justice) will
 be the Convenor of the Commission.
 GRIEVANCES AGAINST JUDICIARY
 Department of Justice receives online/off line grievances from public
@@ -339,45 +338,35 @@ faq_tool=Tool(
     func=retrieve_faq,
     description="Provides Answers to commonly asked questions related to query keyword(s)"
 )
-from langchain.prompts import PromptTemplate
-from langchain.chains import LLMChain
-from langchain.memory import ConversationBufferMemory
-
-prompt = PromptTemplate(
-    template="""
-    You are a highly specialized legal assistant with deep knowledge of the Indian Penal Code (IPC). 
-    Your primary task is to retrieve and summarize legal information accurately from the IPC.pdf document provided to you. 
-    Your responses should be highly specific, fact-based, and free from any speculation or hallucinations. 
-    Always cite the exact section from the IPC when providing an answer. 
-    If the information is not available in the document, clearly state that and do not make any assumptions.
-    Respond to user queries and engage in conversation to resolve their query.
-
-    Example task: "What is the punishment for theft according to the IPC?"
-    Example response: "According to Section 379 of the IPC, the punishment for theft is imprisonment of either description for a term which may extend to three years, or with fine, or with both."
+# Language model setup using Google Gemini
+llm = ChatGoogleGenerativeAI(
+    model="gemini-1.5-pro",
+    temperature=1,
+    max_tokens=None,
+    timeout=None,
+    max_retries=2,
+    prompt_template="""
+    You are JobAI.You process user input and generate human-like responses to assist with job searching and application processes. Tasks include:
+(provide links where ever required)
 
-    History: {history}
+- Searching for job openings based on user criteria
+- Providing job descriptions and requirements
+- Analyzing user resumes and offering suggestions for improvement
+- Generating cover letters tailored to specific job openings
+- Offering interview preparation assistance
 
-    User: {query}
-
-    Response:
+Respond to user queries and engage in conversation to guide them through the job application process. Utilize context and understanding to provide accurate and helpful responses
+   {{history}}
+   User Query: {{query}}
+   
     """,
-    input_variables=["history", "query"]
 )
 
-llm = HuggingFaceEndpoint(
-    repo_id="mistralai/Mistral-Nemo-Instruct-2407",
-    task="text-generation",
-    timeout=10
-)
-
-
-llm_chain = LLMChain(prompt=prompt, llm=llm)
-
 agent_tools = [ipc_tool, crpc_tool, doj_tool, faq_tool]
 
 agent = initialize_agent(
     tools=agent_tools,
-    llm=llm_chain,
+    llm=llm,
     agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
     verbose=True,
     return_intermediate_steps=True,