Update app.py

app.py

@@ -73,9 +73,40 @@ def update_button_state(text):
     else:
         return gr.update(interactive=False)
 
+
+# Assuming 'table_names' is a list of table names fetched from the schema, e.g., ['Reports', 'Tasks', 'Clients']
+
+# Function to fetch table names from schema (you may already have this in your app)
+def get_table_names():
+    # Replace this with actual schema fetching logic if needed
+    return ["Reports", "Tasks", "Clients", "Agents", "Learners", "Classes", "Events", "Assesments", "Progressions", "Deliveries", "Agent Assignments", "Agent Availability", "Agent Work History"]
+
+# Function to update the query textbox when a button is clicked
+def insert_table_name(current_text, table_name):
+    # Add the table name to the current text
+    return current_text + " " + table_name
+
+# Fetch table names from schema
+table_names = get_table_names()
+
+
 # Gradio interface setup
 with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Ubuntu"), "Arial", "sans-serif"], text_size='sm')) as ydcoza_face:
+
     text_input = gr.Textbox(lines=2, label="Text Query")
+
+
+# Adding buttons below the Textbox in Gradio
+with gr.Blocks() as ydcoza_face:
+
+
+    text_input = gr.Textbox(lines=2, label="Text Query")
+    
+    # Dynamically create buttons for each table
+    with gr.Row():
+        for table in table_names:
+            gr.Button(table, size="small", elem_classes="ydcoza-small-button").click(fn=lambda current_text, t=table: insert_table_name(current_text, t), inputs=text_input, outputs=text_input)
+
     examples = gr.Examples(examples=[
         "I'm trying to figure out which agents are the busiest. Can you show me like the top few agents who have a lot on their plate? I'd like to see their names and maybe their contact info if we have it.", 
         "I need to get an overview of our classes. Could you pull up a list that shows what each class is about and which client it's for? Oh, and it would be great to know how many students are in each class. Maybe order it so the biggest classes are at the top?",
@@ -140,3 +171,4 @@ if __name__ == "__main__":
     # ydcoza_face.launch(auth=auth_users, auth_message="Demo Login, Username: admin & Password: 1234")
     # run gradio deploy in Terminal
     ydcoza_face.launch()
+