Update app.py

app.py

@@ -20,12 +20,12 @@ article = """
 <img src='http://www.simoninithomas.com/test/gandalf.jpg', alt="Gandalf"/>"""
 theme="huggingface"
 
-context_setup = prompt
-context = context_setup
-interlocutor_names = ["Player", npc_name]
+
+
+
 
 # Builds the prompt from what previously happened 
-def build_prompt(conversation, context):
+def build_prompt(conversation, context, interlocutor_names):
   prompt = context + "\n"
   for user_msg, resp_msg in conversation:
       line = "\n- " + interlocutor_names[0] + ":" + user_msg
@@ -36,19 +36,20 @@ def build_prompt(conversation, context):
   return prompt
 
 # Recognize what the model said, if it used the correct format
-def clean_chat_output(txt, prompt):
+def clean_chat_output(txt, prompt, interlocutor_names):
   delimiter = "\n- "+interlocutor_names[0]
   output = txt.replace(prompt, '')
   output = output[:output.find(delimiter)]
   return output
 
 
-def chat(top_p, temperature, max_new_tokens, message):
+def chat(message, npc_name, prompt, top_p, temperature, max_new_tokens):
+    interlocutor_names = ["Player", npc_name]
     history = gr.get_state() or []
     history.append((message, ""))
     gr.set_state(history)
     conversation = history
-    prompt = build_prompt(conversation, context)
+    prompt = build_prompt(conversation, context, interlocutor_names)
     
     # Build JSON
     json_ = {"inputs": prompt,
@@ -62,7 +63,7 @@ def chat(top_p, temperature, max_new_tokens, message):
        
     output = query(json_)
     output = output[0]['generated_text']
-    answer = clean_chat_output(output, prompt)
+    answer = clean_chat_output(output, prompt, interlocutor_names)
     response = answer
     history[-1] = (message, response)
     gr.set_state(history)