Update app.py

app.py

@@ -509,7 +509,7 @@ def bind_llm(llm, tools,prompt_template):
 # Define input and output models using Pydantic
 class QueryInput(BaseModel):
     question: str = Field(
-        description="The question to be answered by appropriate tool. Please follow the instructions. For API tool, do not send the question as it is. Please send the ASN id.")# Invoke datavisulaization tool by processing the user question and send two inputs to the tool. One input will be the warehouse name and another input to the tool will be the entire user_question itself. Please join those two strings and send them as a single input string with ':' as delimiter")
+        description="The question to be answered by appropriate tool. Please follow the instructions. ")
     # config: dict = Field(default={}, description="Optional configuration for the database query.")
 
 
@@ -522,27 +522,7 @@ class QueryOutput(BaseModel):
 # Wrap the function with StructuredTool for better parameter handling
 tools = [
     
-    StructuredTool(
-        func=get_ASN_data,
-        name="APIData",
-        args_schema=QueryInput,
-        output_schema=QueryOutput,
-        description="Tool to get details of ASN api. ASN id will be in the input with the format of first three letters as ASN and it is followed by 11 digit numeral. Pass only the id as input.  Do not send the complete user question to the tool. If there are any other queries related to ASN without ASN id, please use the document tool."
-    ),
-    StructuredTool(
-        func=document_data_tool,
-        name="DocumentData",
-        args_schema=QueryInput,
-        output_schema=QueryOutput,
-        description="You are an AI assistant trained to help with warehouse management questions based on a detailed document about our WMS. The document covers various processes such as ASN handling, purchase orders, cross docking, appointment scheduling for shipments, and yard management. Please provide a complete and concise response within 200 words and Ensure that the response is not truncated and covers the essential points. "
-    ),
-    StructuredTool(
-        func=database_tool,
-        name="DatabaseQuery",
-        args_schema=QueryInput,
-        output_schema=QueryOutput,
-        description="Tool to query the database based on structured input."
-    ),
+    
     StructuredTool(
         func=inventory_report,
         name="dataVisualization",
@@ -561,11 +541,7 @@ tools = [
     )
 ]
 
-prompt_template = f"""You are an assistant that helps with database queries, API information, and document retrieval.  Your job is to provide clear, complete, and detailed responses to the following queries. Please give the output response in an user friendly way and remove "**" from the response. For example, document related queries can be answered in a clear and concise way with numbering and not as a paragraph. Database related queries should be answered with proper indentation and use numbering for the rows. ASN id related queries should be answered with proper indentation and use numbering for the rows.
-
-For ASN id related questions, if the user specifies an ASN id, provide the information from the api tool. Pass only the id as input to the tool. Do not pass the entire question as input to the tool. If the details are not found, say it in a clear and concise way.
-You are an AI assistant trained to help with warehouse management questions based on a detailed document about our WMS. The document covers various processes such as ASN handling, purchase orders, cross docking, appointment scheduling for shipments, and yard management. Please provide a complete and concise response within 200 words and Ensure that the response is not truncated and covers the essential points. When answering, focus on providing actionable insights and clear explanations related to the specific query.  Please remove "**" from the response.
-For SQL database-related questions, only use the fields available in the warehouse schema, including tables such as customer_master, efs_company_master, efs_group_company_master, efs_region_master, party_address_detail, wms_warehouse_master.
+prompt_template = f"""You are an assistant that helps with document retrieval, excel analysis and visual analysis.  Your job is to provide clear, complete, and detailed responses to the following queries. Please give the output response in an user friendly way and remove "**" from the response. For example, document related queries can be answered in a clear and concise way with numbering and not as a paragraph. 
 For datavisualization, user will ask for inventory report of a particular warehouse. Your job is to return the image path to chat interface and display the image as output.
 If the proper response is generated, return the response back. Do not iterate again and again.
 {{agent_scratchpad}}
@@ -1106,11 +1082,8 @@ def add_to_redmindgpt(title, summary):
         # Add the new tool to the beginning
         tools.insert(0, run_time_tool)
                
-        prompt_template = f"""You are an assistant that helps with database queries, API information, and document retrieval.  Your job is to provide clear, complete, and detailed responses to the following queries. Please give the output response in an user friendly way and remove "**" from the response. For example, document related queries can be answered in a clear and concise way with numbering and not as a paragraph. Database related queries should be answered with proper indentation and use numbering for the rows. ASN id related queries should be answered with proper indentation and use numbering for the rows.
+        prompt_template = f"""You are an assistant that helps with database queries, API information, and document retrieval.  Your job is to provide clear, complete, and detailed responses to the following queries. Please give the output response in an user friendly way and remove "**" from the response. For example, document related queries can be answered in a clear and concise way with numbering and not as a paragraph.
         {run_time_tool_summary}
-        For ASN id related questions, if the user specifies an ASN id, provide the information from the api tool. Pass only the id as input to the tool. Do not pass the entire question as input to the tool. If the details are not found, say it in a clear and concise way.
-        You are an AI assistant trained to help with warehouse management questions based on a detailed document about our WMS. The document covers various processes such as ASN handling, purchase orders, cross docking, appointment scheduling for shipments, and yard management. Please provide a complete and concise response within 200 words and Ensure that the response is not truncated and covers the essential points. When answering, focus on providing actionable insights and clear explanations related to the specific query.  Please remove "**" from the response.
-        For SQL database-related questions, only use the fields available in the warehouse schema, including tables such as customer_master, efs_company_master, efs_group_company_master, efs_region_master, party_address_detail, wms_warehouse_master.
         For datavisualization, user will ask for inventory report of a particular warehouse. Your job is to return the image path to chat interface and display the image as output.
     
         {{agent_scratchpad}}
@@ -1135,11 +1108,8 @@ def add_to_redmindgpt(title, summary):
         # Add the new tool to the beginning
         tools.insert(0, run_time_excel_tool)
             
-        prompt_template = f"""You are an assistant that helps with database queries, API information, and document retrieval.  Your job is to provide clear, complete, and detailed responses to the following queries. Please give the output response in an user friendly way and remove "**" from the response. For example, document related queries can be answered in a clear and concise way with numbering and not as a paragraph. Database related queries should be answered with proper indentation and use numbering for the rows. ASN id related queries should be answered with proper indentation and use numbering for the rows.
+        prompt_template = f"""You are an assistant that helps with database queries, API information, and document retrieval.  Your job is to provide clear, complete, and detailed responses to the following queries. Please give the output response in an user friendly way and remove "**" from the response. For example, document related queries can be answered in a clear and concise way with numbering and not as a paragraph.
         {run_time_excel_tool_summary}
-        For ASN id related questions, if the user specifies an ASN id, provide the information from the api tool. Pass only the id as input to the tool. Do not pass the entire question as input to the tool. If the details are not found, say it in a clear and concise way.
-        You are an AI assistant trained to help with warehouse management questions based on a detailed document about our WMS. The document covers various processes such as ASN handling, purchase orders, cross docking, appointment scheduling for shipments, and yard management. Please provide a complete and concise response within 200 words and Ensure that the response is not truncated and covers the essential points. When answering, focus on providing actionable insights and clear explanations related to the specific query.  Please remove "**" from the response.
-        For SQL database-related questions, only use the fields available in the warehouse schema, including tables such as customer_master, efs_company_master, efs_group_company_master, efs_region_master, party_address_detail, wms_warehouse_master.
         For datavisualization, user will ask for inventory report of a particular warehouse. Your job is to return the image path to chat interface and display the image as output.
     
         {{agent_scratchpad}}
@@ -1515,9 +1485,6 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
 
     # Buttons for sample queries
     with gr.Row():
-        sample_button = gr.Button("What are the details of ASN24091600002", elem_classes="gr-buttonbig")
-        sample_button1 = gr.Button("What are the active warehouses available", elem_classes="gr-buttonbig")
-        sample_button2 = gr.Button("Explain Pre-Receiving Yard Management", elem_classes="gr-buttonbig")
         sample_button3 = gr.Button("can you generate a doughnut chart with item name and quantities for warehouse DUBAIWARE", elem_classes="gr-buttonbig")
         sample_button4 = gr.Button("Analyze item name & quantity for different customers in a stacked bar chart for the warehouse DUBAIWARE & send email to [email protected]", elem_classes="gr-button")
 
@@ -1568,9 +1535,6 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     submit_feedback_button.click(submit_feedback, [feedback_textbox, chatbot], [chatbot, feedback_textbox, submit_feedback_button])
     submit_feedback_button.click(lambda x: gr.update(value=''), [], [feedback_textbox])
 
-    sample_button.click(handle_query, [sample_button, chatbot], [chatbot])
-    sample_button1.click(handle_query, [sample_button1, chatbot], [chatbot])
-    sample_button2.click(handle_query, [sample_button2, chatbot], [chatbot])
     sample_button3.click(handle_query, [sample_button3, chatbot], [chatbot])
     sample_button4.click(handle_query, [sample_button4, chatbot], [chatbot])