Update app.py

app.py

@@ -14,17 +14,31 @@ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 #         prompt += f" {bot_response}</s> "
 #     prompt += f"[INST] {message} [/INST]"
 #     return prompt
-def format_prompt(message, history):
+# def format_prompt(message, history):
+#     prompt = "<s>"
+#     # Start the conversation with a system message
+#     prompt += "[INST] You are a Travel Companion Chatbot that helps users plan trips by suggesting transport, sightseeing stops, and accommodations based on their preferences. Please assist the user by asking what they need to know. [/INST]"
+    
+#     # Only append the user message, without the historical responses or examples
+#     prompt += f"[INST] {message} [/INST]"
+    
+#     return prompt
+def format_prompt(message):
     prompt = "<s>"
-    # Start the conversation with a system message
-    prompt += "[INST] You are a Travel Companion Chatbot that helps users plan trips by suggesting transport, sightseeing stops, and accommodations based on their preferences. Please assist the user by asking what they need to know. [/INST]"
+    # 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]"
     
-    # Only append the user message, without the historical responses or examples
-    prompt += f"[INST] {message} [/INST]"
+    # Include the user's message as the input to be processed
+    prompt += f"{message}"
     
     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