Add query op and alarm table

app.py

@@ -96,7 +96,7 @@ def main():
     path = r"OPPI_shift.db"       # \OPPI_down.db"
     db1 = SQLDatabase.from_uri(f"sqlite:///{path}", include_tables=['ShiftDownTimeDetails'],sample_rows_in_table_info=0)
     db2 = SQLDatabase.from_uri(f"sqlite:///{path}", include_tables=['ShiftProductionDetails'],sample_rows_in_table_info=0)
-    db3 = SQLDatabase.from_uri(f"sqlite:///{path}", include_tables=['ShiftDownTimeDetails','ShiftProductionDetails'],sample_rows_in_table_info=0)
+    db3 = SQLDatabase.from_uri(f"sqlite:///{path}", include_tables=['ShiftDownTimeDetails','ShiftProductionDetails','Focas_AlarmHistory'],sample_rows_in_table_info=0)
     
     down_chain = chain(db=db1, llm=gemini)
     prod_chain = chain(db=db2, llm=gemini)
@@ -115,22 +115,35 @@ def main():
             ("human", "{question}"+table_info)
         ]
     )
+    
+    examples = [
+        "calculate total Prod quantity in Second Shift for 2024",
+        "Calculate total accepted parts in shift 2 for 2024",
+        "How many accepted parts were produced in October 2023 in each machine",
+        "How likely is the Turrent index aborted alarm expected on machine k-1",
+        "List all the distinct reasons behind DownTime in machine K-2",
+        "Calculate the total Downtime experienced by machine K-8 due to the reason of No Shift",
+        "What was the most common reason for Downtime in the year 2023?",
+        "Calculate the average downtime for Machine M-2 in for every month in later half of 2023",
+        "return all the reasons for Downcategory in Nov and dec on machine L-7 in 3rd shift",
+    ]
     sql_toolkit = SQLDatabaseToolkit(db=db3, llm=agent_llm)
     agent = create_sql_agent(
         toolkit=sql_toolkit,
         llm=agent_llm,
         agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
         verbose=True,
-        agent_executor_kwargs={"handle_parsing_errors":True}
+        agent_executor_kwargs={"handle_parsing_errors":True,  "return_intermediate_steps": True}
     )
 
     def echo3(message, history):
         answer = agent.invoke(prompt_agent.format_prompt(question=message))
-        return answer['output']
+        final_answer = f"Final Query:- {list(answer['intermediate_steps'][1][0])[-2][1].split('Action Input: ')[-1]}\n\nAnswer:- {answer['output']}"
+        return final_answer
     
     downtime = gr.ChatInterface(fn=echo1, title="SQL-Chatbot", description="Q/A on Downtime details table")
     production = gr.ChatInterface(fn=echo2, title="SQL-Chatbot", description="Q/A on Production details table")
-    agent_tab = gr.ChatInterface(fn=echo3, title="SQL-Chatbot", description="General Chatbot with self-thinking capability, more robust to questions.")
+    agent_tab = gr.ChatInterface(fn=echo3, examples=examples, title="SQL-Chatbot", description="General Chatbot with self-thinking capability, more robust to questions.")
     demo = gr.TabbedInterface([agent_tab, downtime, production], ['DB_bot-both tables','ShiftDownTimeDetails', 'ShiftProductionDetails'])
     demo.launch(debug=True, share=True)