corrected the app.py

app.py

@@ -16,46 +16,24 @@ def query(payload):
     return response.json()
 history  = []
 
-# def chat(message,history=[]):
-#     response = query({
-#         "messages": [
-
-#             {
-#                 "role": "user",
-#                 "content": message
-#             }
-#         ],
-#         "model": "meta-llama/Llama-3.1-8B-Instruct:novita"
-#     })
-#     result = response["choices"][0]["message"]
-#     print(result)
-#     history.append({
-#                 "role": "user",
-#                 "content": message
-#             })
-#     history.append({
-#                 "role": "assistant",
-#                 "content": result
-#             })
-#     return result["content"]
-
-def chat(role,message):
+
+def chat(message, history):
+    messages = []
+
+    # Add previous conversation
+    for user_msg, bot_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
+
+    # Add current user message
+    messages.append({"role": "user", "content": message})
+
     response = query({
-        "messages": [
-            {
-                "role": "user",
-                "content": message
-            }
-        ],
-        "model": "meta-llama/Llama-3.1-8B-Instruct:novita"
+        "model": "meta-llama/Llama-3.1-8B-Instruct:novita",
+        "messages": messages
     })
-    print(response)
-    result = response["choices"][0]["message"]
-    print(result)
-    return result["content"]
-
 
-print(chat("user","hi"))
+    return response["choices"][0]["message"]["content"]
 
 gr.ChatInterface(
     fn=chat,