Update app_template.py

app_template.py

@@ -11,11 +11,11 @@ def build_input_prompt(message, chatbot, system_prompt):
     """
     Constructs the input prompt string from the chatbot interactions and the current message.
     """
-    input_prompt = "<|system|>\n" + system_prompt + "</s>\n<|user|>\n"
+    input_prompt = "\n" + system_prompt + "</s>\n\n"
     for interaction in chatbot:
-        input_prompt = input_prompt + str(interaction[0]) + "</s>\n<|assistant|>\n" + str(interaction[1]) + "\n</s>\n<|user|>\n"
+        input_prompt = input_prompt + str(interaction[0]) + "</s>\n\n" + str(interaction[1]) + "\n</s>\n\n"
 
-    input_prompt = input_prompt + str(message) + "</s>\n<|assistant|>"
+    input_prompt = input_prompt + str(message) + "</s>\n"
     return input_prompt
 
 
@@ -55,20 +55,31 @@ def predict_beta(message, chatbot=[], system_prompt=""):
 
 def test_preview_chatbot(message, history):
     response = predict_beta(message, history, SYSTEM_PROMPT)
-    text_start = response.rfind("<|assistant|>", ) + len("<|assistant|>")
+    text_start = response.rfind("", ) + len("")
     response = response[text_start:]
     return response
 
 
+# Define your custom CSS here
+css = """
+/* Add your custom CSS styles here */
+"""
+
+# Define your theme here
+theme = "ParityError/Interstellar"
+
 welcome_preview_message = f"""
 Welcome to **{TITLE}**! Say something like: 
-
 "{EXAMPLE_INPUT}"
 """
 
 chatbot_preview = gr.Chatbot(layout="panel", value=[(None, welcome_preview_message)])
 textbox_preview = gr.Textbox(scale=7, container=False, value=EXAMPLE_INPUT)
 
-demo = gr.ChatInterface(test_preview_chatbot, chatbot=chatbot_preview, textbox=textbox_preview)
+# Modify this part to include your theme and CSS
+with gr.Blocks(css=css, theme=theme) as demo:
+    demo.add_component(chatbot_preview)
+    demo.add_component(textbox_preview)
+    demo.add_component(gr.Button("Submit"))
 
-demo.launch()
+demo.launch()