Update app.py

app.py

@@ -5,13 +5,32 @@ client = InferenceClient(
     "mistralai/Mixtral-8x7B-Instruct-v0.1"
 )
 
+# Formats the prompt to hold all of the past messages
+def format_prompt_basic(message, history):
+    prompt = "<s>"
+
+    # String to add before every prompt
+    prompt_prefix = ""   #"Please correct the grammar in the following sentence:"
+    prompt_template = "[INST] " + prompt_prefix + " {} [/INST]"
+    
+
+    # Iterates through every past user input and response to be added to the prompt
+    for user_prompt, bot_response in history:
+        prompt += prompt_template.format(user_prompt)
+        prompt += f" {bot_response}</s> "
+        
+    prompt += prompt_template.format(message)
+    print("\nPROMPT: \n\t" + prompt)
+
+    return prompt
+
 # Formats the prompt to hold all of the past messages
 def format_prompt(message, history):
     prompt = "<s>"
 
     # String to add before every prompt
     prompt_prefix = "Correct any grammatical errors in the following sentence and provide the corrected version:\n\nSentence: "
-    prompt_template = "[INST] " + prompt_prefix + ' "{}" [/INST]'
+    prompt_template = "[INST] " + prompt_prefix + ' {} [/INST]'
     
     
     myList = [["It is my friends house in England.", "It is my friend's house in England."], ["Every girl must bring their books to school.", "Every girl must bring her books to school."]]