Update app.py

app.py

@@ -1,3 +1,4 @@
+import gradio as gr
 from smolagents import CodeAgent, InferenceClientModel
 from tools.party_menu import suggest_menu
 from tools.party_song import suggest_party_music
@@ -11,6 +12,18 @@ agent = CodeAgent(
     tools=[FinalAnswerTool(), suggest_menu, suggest_party_music, estimate_preparation_time],
 )
 
+# 定义 Gradio 接口
+def chat_fn(message, history):
+    result = agent.run(message)
+    history = history + [(message, result)]
+    return history, history
+
+demo = gr.ChatInterface(
+    fn=chat_fn,
+    chatbot=gr.Chatbot(),
+    textbox=gr.Textbox(placeholder="Ask Alfred to help plan your party..."),
+    title="Alfred Party Planner Agent",
+)
+
 if __name__ == "__main__":
-    from Gradio_UI import GradioUI
-    GradioUI(agent).launch()
+    demo.launch()