Update app.py

app.py

@@ -5,10 +5,8 @@ from smolagents import HfApiModel
 
 
 model = HfApiModel(model_id="mistralai/Mixtral-8x7B-Instruct-v0.1", token=os.environ.get("HF_TOKEN"))
-
-def chat(prompt, history):
-    data = [
-        {
+system_data = [
+    {
             "role":"system",
             "content":[
                 {
@@ -16,7 +14,11 @@ def chat(prompt, history):
                     "text": "You are a doctor who specializes on helping patients with addiction issues"
                 }
             ]
-        },
+        }
+]
+
+def get_user_data(prompt: str):
+    return [
         {
             "role":"user",
             "content":[
@@ -27,8 +29,24 @@ def chat(prompt, history):
             ]
         }
     ]
+
+def get_history(history):
+    mod_history = []
+    for i in history:
+        d = {}
+        d["role"] = i["role"]
+        d["content"] = [
+            {
+                "type": "text",
+                "text": i["content"]
+            }
+        ]
+        mod_history.append(d)
+    return mod_history
+
+def chat(prompt, history):
     
-    return model(data).content
+    return model(system_data + get_history(history)+ get_user_data(prompt)).content
 
 demo = gr.ChatInterface(chat, title="ArunGPT",theme = gr.themes.Soft(), description="Hello this is chatbot is created for only educational purpose and is powered by mistral 8x 7b model").queue()