Update app.py

app.py

@@ -6,22 +6,20 @@ client = InferenceClient(
     "meta-llama/Meta-Llama-3-8B-Instruct",
 )
 
-def chat_llama(chat_history):
+def chat_llama(chat_historyV2):
     chat_completion = client.chat_completion(
-        messages=chat_history,
+        messages=chat_historyV2,
         max_tokens=500,
     )
-    chat_history.append({"role": "assistant", "content": chat_completion.choices[0].message.content})
-    print(chat_history)
-    return chat_history
+    chat_historyV2.append({"role": "assistant", "content": chat_completion.choices[0].message.content})
+    return chat_historyV2
 
 def chat_mem(message, chat_history):
-    print(len(chat_history))
-    chat_history_role = [{"role": "system", "content": "You are a helpful assistant."},]
-    if chat_history != []:
-        for i in range(len(chat_history)):
-            chat_history_role.append({"role": "user", "content": chat_history[i][0]})
-            chat_history_role.append({"role": "assistant", "content": chat_history[i][1]})
+    chat_history_role = [{"role": "system", "content": "You are a helpful assistant."}]
+    if chat_history:
+        for user_message, assistant_message in chat_history:
+            chat_history_role.append({"role": "user", "content": user_message})
+            chat_history_role.append({"role": "assistant", "content": assistant_message})
     chat_history_role.append({"role": "user", "content": message})
     
     chat_completion = client.chat_completion(
@@ -29,30 +27,28 @@ def chat_mem(message, chat_history):
         max_tokens=500,
     )
     chat_history_role.append({"role": "assistant", "content": chat_completion.choices[0].message.content})
-    print(chat_history_role)
 
-    modified = map(lambda x: x["content"], chat_history_role)
-    a = list(modified)
-    chat_history = [(a[i*2+1], a[i*2+2]) for i in range(len(a)//2)]
+    # Convert chat history role to the format used in chat history
+    chat_history = [(chat_history_role[i*2+1]['content'], chat_history_role[i*2+2]['content']) for i in range(len(chat_history_role)//2)]
 
     return "", chat_history
 
 def process_json(json_input):
     try:
-        chat_history = json.loads(json_input)
-        if not isinstance(chat_history, list):
+        chat_historyV2 = json.loads(json_input)
+        if not isinstance(chat_historyV2, list):
             raise ValueError("Input should be a list of message dictionaries.")
     except (json.JSONDecodeError, ValueError) as e:
         return f"Error parsing JSON: {str(e)}", ""
     
-    chat_history = chat_llama(chat_history)
-    return json.dumps(chat_history, indent=2), ""
+    chat_historyV2 = chat_llama(chat_historyV2)
+    return json.dumps(chat_historyV2, indent=2), ""
 
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             chatbot = gr.Chatbot()
-            msg = gr.Textbox(interactive=True, )
+            msg = gr.Textbox(interactive=True)
             with gr.Row():
                 clear = gr.ClearButton([msg, chatbot], icon="https://img.icons8.com/?size=100&id=Xnx8cxDef16O&format=png&color=000000")
                 send_btn = gr.Button("Send", variant='primary', icon="https://img.icons8.com/?size=100&id=g8ltXTwIfJ1n&format=png&color=000000")