updated respond fns with tab_name

app.py

@@ -47,15 +47,16 @@ def format_chat_prompt(message, chat_history, max_convo_length):
     return prompt
 
 def gpt_respond(tab_name, message, chat_history, max_convo_length = 10):
-        formatted_prompt = format_chat_prompt(message, chat_history, max_convo_length)
-        print('Prompt + Context:')
-        print(formatted_prompt)
-        bot_message = chat(system_prompt = f'''Generate the output only for the assistant. Please output any <{tab_name}> in the following sentence one per line without any additional text.''',
-                           user_prompt = formatted_prompt)
-        chat_history.append((message, bot_message))
-        return "", chat_history
+    formatted_prompt = format_chat_prompt(message, chat_history, max_convo_length)
+    print('Prompt + Context:')
+    print(formatted_prompt)
+    bot_message = chat(system_prompt = f'''Generate the output only for the assistant. Please output any <{tab_name}> in the following sentence one per line without any additional text.''',
+                       user_prompt = formatted_prompt)
+    chat_history.append((message, bot_message))
+    return "", chat_history
 
-def vicuna_respond(message, chat_history):
+def vicuna_respond(tab_name, message, chat_history):
+    # formatted_prompt = f'''Generate the output only for the assistant. Please output any <{tab_name}> in the following sentence one per line without any additional text: <{message}>'''
     input_ids = vicuna_tokenizer.encode(message, return_tensors="pt")
     output_ids = vicuna_model.generate(input_ids, max_length=50, num_beams=5, no_repeat_ngram_size=2)
     bot_message = vicuna_tokenizer.decode(output_ids[0], skip_special_tokens=True)
@@ -64,7 +65,8 @@ def vicuna_respond(message, chat_history):
     time.sleep(2)
     return "", chat_history
 
-def llama_respond(message, chat_history):
+def llama_respond(tab_name, message, chat_history):
+    # formatted_prompt = f'''Generate the output only for the assistant. Please output any <{tab_name}> in the following sentence one per line without any additional text: <{message}>'''
     input_ids = llama_tokenizer.encode(message, return_tensors="pt")
     output_ids = llama_model.generate(input_ids, max_length=50, num_beams=5, no_repeat_ngram_size=2)
     bot_message = llama_tokenizer.decode(output_ids[0], skip_special_tokens=True)
@@ -88,9 +90,9 @@ def interface():
         textbox_prompt = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
         with gr.Row():
             api_key_input = gr.Textbox(label="Open AI Key", placeholder="Enter your Openai key here", type="password")
-            api_key_btn = gr.Button(label="Submit Api Key", scale=0)
+            api_key_btn = gr.Button(value="Submit Key", scale=0)
         tab_name = gr.Dropdown(["Noun", "Determiner", "Noun phrase", "Verb phrase", "Dependent clause", "T-units"], label="Linguistic Entity")
-        btn = gr.Button(label="Submit")
+        btn = gr.Button(value="Submit")
 
         # prompt = template_single.format(tab_name, textbox_prompt)