cleanup

app.py

@@ -53,18 +53,7 @@ def sql_engine_tool(query: str) -> str:
         query: The query to perform. This should be correct SQL.
     """
     output = ""
-    print("debug sql_engine_tool")
-    print(engine)
     with engine.begin() as con:
-        print(con.connection)
-        print(metadata_objects.tables.keys())
-        result = con.execute(
-            text(
-                "SELECT name FROM sqlite_master WHERE type='table' AND name='receipts'"
-            )
-        )
-        print("tables available:", result.fetchone())
-
         rows = con.execute(text(query))
         for row in rows:
             output += "\n" + str(row)
@@ -72,7 +61,6 @@ def sql_engine_tool(query: str) -> str:
 
 
 def init_db(engine):
-
     metadata_obj = MetaData()
 
     def insert_rows_into_table(rows, table, engine=engine):
@@ -109,17 +97,13 @@ def init_db(engine):
         },
     ]
     insert_rows_into_table(rows, receipts)
-    with engine.begin() as conn:
-        print("SELECT test", conn.execute(text("SELECT * FROM receipts")).fetchall())
-    print("init_db debug")
-    print(engine)
-    print()
-    return engine, metadata_obj
+
+    return engine
 
 
 if __name__ == "__main__":
     engine = create_engine("sqlite:///:localhost:")
-    engine, metadata_objects = init_db(engine)
+    engine = init_db(engine)
     model = HfApiModel(
         model_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
         token=os.getenv("my_first_agents_hf_tokens"),
@@ -131,10 +115,8 @@ if __name__ == "__main__":
         max_steps=10,
         verbosity_level=1,
     )
-    # GradioUI(agent).launch()
 
     def enter_message(new_message, conversation_history):
-
         conversation_history.append(gr.ChatMessage(role="user", content=new_message))
         yield "", conversation_history
         for msg in stream_to_gradio(agent, new_message):
@@ -143,7 +125,7 @@ if __name__ == "__main__":
 
     with gr.Blocks() as b:
         chatbot = gr.Chatbot(type="messages", height=1000)
-        textbox = gr.Textbox()
+        textbox = gr.Textbox(lines=3)
         button = gr.Button("reply")
         button.click(enter_message, [textbox, chatbot], [textbox, chatbot])
-    b.launch(debug=True)
+    b.launch()