Update app.py

app.py

@@ -125,13 +125,13 @@ def main():
     )
 
     def echo3(message, history):
-        answer = agent.invoke(prompt.format_prompt(question=message))
+        answer = agent.invoke(prompt_agent.format_prompt(question=message))
         return answer['output']
     
     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 = gr.ChatInterface(fn=echo3, title="SQL-Chatbot", description="General Chatbot with self-thinking capability, more robust to questions.")
-    demo = gr.TabbedInterface([agent, downtime, production], ['DB_bot-both tables','ShiftDownTimeDetails', 'ShiftProductionDetails'])
+    agent_tab = gr.ChatInterface(fn=echo3, 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)
 
 if __name__ == "__main__":