level 2

app.py

@@ -98,6 +98,22 @@ def init_db(engine):
     ]
     insert_rows_into_table(rows, receipts)
 
+    table_name = "waiters"
+    waiters = Table(
+        table_name,
+        metadata_obj,
+        Column("receipt_id", Integer, primary_key=True),
+        Column("waiter_name", String(16), primary_key=True),
+    )
+    metadata_obj.create_all(engine)
+
+    rows = [
+        {"receipt_id": 1, "waiter_name": "Corey Johnson"},
+        {"receipt_id": 2, "waiter_name": "Michael Watts"},
+        {"receipt_id": 3, "waiter_name": "Michael Watts"},
+        {"receipt_id": 4, "waiter_name": "Margaret James"},
+    ]
+    insert_rows_into_table(rows, waiters)
     return engine
 
 
@@ -105,7 +121,8 @@ if __name__ == "__main__":
     engine = create_engine("sqlite:///:localhost:")
     engine = init_db(engine)
     model = HfApiModel(
-        model_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
+        model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+        # model_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
         token=os.getenv("my_first_agents_hf_tokens"),
     )
 
@@ -125,7 +142,7 @@ if __name__ == "__main__":
 
     with gr.Blocks() as b:
         chatbot = gr.Chatbot(type="messages", height=1000)
-        textbox = gr.Textbox(lines=3)
+        textbox = gr.Textbox(lines=3, label="")
         button = gr.Button("reply")
         button.click(enter_message, [textbox, chatbot], [textbox, chatbot])
     b.launch()