Update app.py

app.py

@@ -106,10 +106,10 @@ def ask_ds(message, history):
         similar_documents.append((file, similarity))
 
     similar_documents.sort(key=lambda x: x[1], reverse=False)
-    top_3 = similar_documents[:3]
+    top_docs = similar_documents[:5]
 
     similar_content = ''
-    for file, _ in top_3:
+    for file, _ in top_docs:
         similar_content += extractions[file]['content'] + '\n'
 
     # Invoke
@@ -119,13 +119,25 @@ def ask_ds(message, history):
             {
                 "anthropic_version": "bedrock-2023-05-31",
                 "max_tokens": 4096,
-                "system": f"""You are a helpful, excited assistant that answers questions about certain provided documents. 
-                <Task>
+                "system": f"""Here is some relevant information that may help answer the user's upcoming question:
+
+                <relevant_information>
+                {similar_content}
+                </relevant_information>
+                
+                The user's question is:
+                <question>{question}</question>
+
+                Please carefully review the relevant information provided above. 
+
                 Your task is to review the provided relevant information and answer the user's question to the best of your ability. 
-                Try to use only the information in the document to answer. Refrain from saying things like 'According to the relevant information provided'.
+                Aim to use information from the relevant information section to directly address the question asked, and refrain from saying 
+                things like 'According to the relevant information provided'.
 
-                Format your output nicely with sentences that are not too long. You should prefer lists or bullet points but only when applicable.
-                Begin by thanking the user for their question, and at the end of your answer, say "Thank you for using Ask Dane Street!"
+                Format your output nicely with sentences that are not too long, in a professional and kind tone. You should prefer lists or 
+                bullet points when applicable. Begin by thanking the user for their question, and at the end of your answer, say "Thank you for using Ask Dane Street!"
+                Remember, aim to only use information from the relevant information section in your response, without explicitly referring 
+                to that section. Return your answer immediately and without preamble. 
                 </Task>
                 
                 <Relevant Information>
@@ -163,7 +175,7 @@ def ask_ds(message, history):
 
     # Print relevant files
     output = '\n\nCheck out the following documents for more information:\n'
-    for file, sim in top_3:
+    for file, sim in top_docs:
         output += f"\n{file.replace('.txt', '.pdf')}"
 
     yield all_text + output
@@ -174,5 +186,5 @@ bedrock_client = create_bedrock_client()
 s3_client = create_s3_client()
 extractions = read_json_from_s3()
 
-demo = gr.ChatInterface(fn=ask_ds, title="AskDS_HR", multimodal=False, chatbot=gr.Chatbot(value=[(None, "Welcome to Ask HR at Dane Street! Whether you're new to the team or just looking for some quick information, I'm here to guide you through our company's literature. From Benefits and Payroll queries to understanding our Policies or Paylocity, simply ask your question and I'll provide you with the most relevant information I can.\n\nCurious about your 401(k) account changes? Or maybe you need details on the 2023 Benefits Guide? I can help with that and much more! Don't hesitate to ask, and thank you for making Dane Street an amazing place to work. If there’s something specific you'd like to know, please type your question below, and I will do my best to assist you.")],),theme=theme)
+demo = gr.ChatInterface(fn=ask_ds, title="Ask DS", multimodal=False, chatbot=gr.Chatbot(value=[(None, "Welcome to Ask Dane Street! Whether you're new to the team or just looking for some quick information, I'm here to guide you through our company's literature and platform. Simply ask your question, and I'll provide you with the most relevant information I can.")],),theme=theme)
 demo.launch()