Update app.py

app.py

@@ -8,7 +8,7 @@ model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 client = InferenceClient(model)
 
 # Embedded system prompt
-system_prompt_text = "You are a smart and helpful co-worker of Thailand based multi-national company PTT, and PTTEP. You help with any kind of request and provide a detailed answer to the question. But if you are asked about something unethical or dangerous, you must refuse and provide a safe and respectful way to handle that"
+system_prompt_text = "You are a smart and helpful co-worker of Thailand based multi-national company PTT, and PTTEP. You help with any kind of request and provide a detailed answer to the question. But if you are asked about something unethical or dangerous, you must refuse and provide a safe and respectful way to handle that."
 
 # Read the content of the info.md file
 with open("info.md", "r") as file:
@@ -18,14 +18,13 @@ with open("info.md", "r") as file:
 chunk_size = 2000  # Adjust this size as needed
 info_md_chunks = textwrap.wrap(info_md_content, chunk_size)
 
-def get_relevant_chunk(prompt, chunks):
-    # For simplicity, we just use the first chunk. You can improve this by adding more sophisticated logic.
-    return chunks[0]
+def get_all_chunks(chunks):
+    return "\n\n".join(chunks)
 
-def format_prompt_mixtral(message, history, info_md_content):
+def format_prompt_mixtral(message, history, info_md_chunks):
     prompt = "<s>"
-    relevant_chunk = get_relevant_chunk(message, info_md_content)
-    prompt += f"{relevant_chunk}\n\n"  # Add the relevant chunk of info.md at the beginning
+    all_chunks = get_all_chunks(info_md_chunks)
+    prompt += f"{all_chunks}\n\n"  # Add all chunks of info.md at the beginning
     prompt += f"{system_prompt_text}\n\n"  # Add the system prompt
 
     if history:
@@ -66,7 +65,7 @@ def check_rand(inp, val):
         return gr.Slider(label="Seed", minimum=1, maximum=1111111111111111, value=int(val))
 
 with gr.Blocks() as app:  # Add auth here
-    gr.HTML("""<center><h1 style='font-size:xx-large;'>PTT Chatbot</h1><br><h3>running on Huggingface Inference </h3><br><h7>EXPERIMENTAL""")
+    gr.HTML("""<center><h1 style='font-size:xx-large;'>PTT Chatbot</h1><br><h3>running on Huggingface Inference </h3><br><h7>EXPERIMENTAL</center>""")
     with gr.Row():
         chat = gr.Chatbot(height=500)
     with gr.Group():