Update app.py

app.py

@@ -1,44 +1,33 @@
-
 import os
 import gradio as gr
-from langchain_community.llms import HuggingFaceEndpoint
+from langchain_huggingface import HuggingFaceEndpoint
 from langchain.prompts import PromptTemplate
 
 # Initialize the chatbot
 HF_TOKEN = os.getenv("HF_TOKEN")
 llm = HuggingFaceEndpoint(
-    #repo_id="google/gemma-2b-it",
-    #repo_id="abhinand/gemma-2b-tamil",
-    repo_id="google/gemma-1.1-7b-it",
-    #repo_id="google/gemma-2b",
-    
+    repo_id="google/gemma-1.1-7b-it",  # Update this to your desired model
     task="text-generation",
     max_new_tokens=512,
     top_k=5,
     temperature=0.3,
     repetition_penalty=1.03,
-    
-    huggingfacehub_api_token=HF_TOKEN
+    huggingfacehub_api_token=HF_TOKEN,
+    add_to_git_credential=True  # Add this line to handle git credentials
 )
+
 template = """
 You are a Mental Health Chatbot, your purpose is to provide supportive and non-judgmental guidance to users who are struggling with their mental health. Your goal is to help users identify their concerns, offer resources and coping strategies, and encourage them to seek professional help when needed. If the user symptoms are not related to Mental Health reply that you are a mental health chatbot and have no knowledge about other diseases.
 User Context: {context}
 Question: {question}
-
 If the user symptoms are not related to Mental Health or related to other disease, disability or disorder reply that you are a mental health chatbot and you cannot provide any details on that.
-
 Please respond with a helpful and compassionate answer that addresses the user's concern about their mental health. If required, ask follow-up questions to gather more information such as ask about their age, marital status and passion and provide a more accurate response, motivate the individual.
-
-If the user needs help on any other diseases, disability or disorder which are irrelavent or not reated to their mental health tell them that you are a Mental health chatbot trained for support and guidance.
-
-Only if the user needs to be motivated, then narrate a motivation story with some life quotes and quotes by successful people about life (dont provide the motivation story all the time and at the begining of the conversation)
-
+If the user needs help on any other diseases, disability or disorder which are irrelevant or not related to their mental health tell them that you are a Mental health chatbot trained for support and guidance.
+Only if the user needs to be motivated, then narrate a motivation story with some life quotes and quotes by successful people about life (don't provide the motivation story all the time and at the beginning of the conversation)
 Remember to prioritize the user's well-being and safety. If the user expresses suicidal thoughts or intentions, please respond with immediate support and resources, such as the National Suicide Prevention Lifeline (+91 91529 87821-TALK) in India, or other similar resources in your region. 
-
 Helpful Answer: """
 
-
-QA_CHAIN_PROMPT = PromptTemplate(input_variables=["context", "question"],template=template)
+QA_CHAIN_PROMPT = PromptTemplate(input_variables=["context", "question"], template=template)
 
 def predict(message, history):
     input_prompt = QA_CHAIN_PROMPT.format(question=message, context=history)
@@ -53,5 +42,5 @@ def predict(message, history):
     
     return ai_msg
 
-
-gr.ChatInterface(predict).launch()
+# Launch the Gradio Chat Interface
+gr.ChatInterface(predict).launch()