Update app.py

app.py

@@ -21,9 +21,6 @@ article = """
 theme="huggingface"
 
 
-
-
-"""
 # Builds the prompt from what previously happened 
 def build_prompt(conversation, context, interlocutor_names):
   prompt = context + "\n"
@@ -41,30 +38,27 @@ def clean_chat_output(txt, prompt, interlocutor_names):
   output = txt.replace(prompt, '')
   output = output[:output.find(delimiter)]
   return output
-"""
-
 
 
-def chat(message, npc_name, prompt, top_p, temperature, max_new_tokens, history=[]):
+def chat(message, npc_name, initial_prompt, top_p, temperature, max_new_tokens, history=[]):
     interlocutor_names = ["Player", npc_name]
     
     print("message", message)
     print("npc_name", npc_name)
+    print("initial_prompt", initial_prompt)
     print("top_p", top_p)
     print("temperature", temperature)
     print("max_new_tokens", max_new_tokens)
     print("history", history)
     response = "Test"
-    history.append((message, response))
-    """
-    history = gr.get_state() or []
     history.append((message, ""))
-    gr.set_state(history)
     conversation = history
-    prompt = build_prompt(conversation, context, interlocutor_names)
+    
+    # Build the prompt
+    prompt = build_prompt(conversation, initial_prompt, interlocutor_names)
     
     # Build JSON
-    json_ = {"inputs": prompt,
+    json_req = {"inputs": prompt,
          "parameters":
          {
          "top_p": top_p,
@@ -72,14 +66,14 @@ def chat(message, npc_name, prompt, top_p, temperature, max_new_tokens, history=
         "max_new_tokens": max_new_tokens,
         "return_full_text": False
         }}
-       
-    output = query(json_)
+    
+    # Get the output
+    output = query(json_req)
     output = output[0]['generated_text']
+    
     answer = clean_chat_output(output, prompt, interlocutor_names)
     response = answer
     history[-1] = (message, response)
-    gr.set_state(history)
-    """
     return response, history
 
 
@@ -88,7 +82,7 @@ def chat(message, npc_name, prompt, top_p, temperature, max_new_tokens, history=
 iface = gr.Interface(fn=chat, 
 inputs=[Textbox(label="message"),
         Textbox(label="npc_name"),
-        Textbox(label="prompt"),
+        Textbox(label="initial_prompt"),
         Slider(minimum=0.5, maximum=1, step=0.05, default=0.9, label="top_p"),
         Slider(minimum=0.5, maximum=1.5, step=0.1, default=1.1, label="temperature"),
         Slider(minimum=20, maximum=250, step=10, default=50, label="max_new_tokens"),