Update app.py

app.py

@@ -23,22 +23,16 @@ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 #     prompt += f"[INST] {message} [/INST]"
     
 #     return prompt
-def format_prompt(message):
+def format_prompt(message, history):
     prompt = "<s>"
-    # System message to set the context of the AI's purpose
-    prompt += "[INST] You are a Travel Companion chatbot designed to assist users in planning their trips. When a user provides the source, destination, and the number of days they are planning for a trip, you should respond by:\n"
-    prompt += "- Suggesting the best travel options (bus, train, flight, etc.) with cost-effective choices.\n"
-    prompt += "- Recommending cost-effective hotels and restaurants along the route.\n"
-    prompt += "- Highlighting the best places to visit on the route.\n"
-    prompt += "Please respond directly to the user's input with a detailed plan, without repeating their message. [/INST]"
+    # Start the conversation with a system message
+    prompt += "[INST] You are a Travel Companion chatbot designed to assist users with planning trips. Your role is to provide the best travel options (bus, train, flight) that are cost-effective, suggest the best hotels and restaurants along the route, and recommend the best places to visit based on the user's input. When the user provides the source, destination, and number of days for their trip, you should respond with detailed, cost-effective suggestions. If the user hasn't provided enough details, ask for more information on the trip. [/INST]"
     
-    # Include the user's message as the input to be processed
-    prompt += f"{message}"
+    # Append the user's input message to the prompt
+    prompt += f"[INST] {message} [/INST]"
     
     return prompt
 
-
-
 # Function to generate responses with the AI Dermatologist context
 def generate(
     prompt, history, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0