Update app.py

app.py

@@ -1,54 +1,46 @@
-import gradio as gr
-from smolagents import (CodeAgent, InferenceClientModel)
-from langfuse import get_client
-
-langfuse = get_client()
-
-model = InferenceClientModel()
-agent = CodeAgent(tools=[], model=model, add_base_tools=True)
-
-trace_id = None
-
-def respond(prompt, history):
-    with langfuse.start_as_current_span(
-        name="Smolagent-Trace"):
-        
-        # Run your application here
-        output = agent.run(prompt)
-
-        global trace_id
-        trace_id = langfuse.get_current_trace_id()
-
-    history.append({"role": "assistant", "content": str(output)})
-    return history
-
-def handle_like(data: gr.LikeData):
-    # For demonstration, we map user feedback to a 1 (like) or 0 (dislike)
-    if data.liked:
-        langfuse.create_score(
-            value=1,
-            name="user-feedback",
-            trace_id=trace_id
-        )
-    else:
-        langfuse.create_score(
-            value=0,
-            name="user-feedback",
-            trace_id=trace_id
-        )
-
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot(label="Chat", type="messages")
-    prompt_box = gr.Textbox(placeholder="Type your message...", label="Your message")
-
-    # When the user presses 'Enter' on the prompt, we run 'respond'
-    prompt_box.submit(
-        fn=respond,
-        inputs=[prompt_box, chatbot],
-        outputs=chatbot
-    )
-
-    # When the user clicks a 'like' button on a message, we run 'handle_like'
-    chatbot.like(handle_like, None, None)
-
-demo.launch()
+import yaml
+import os
+from smolagents import GradioUI, CodeAgent, HfApiModel
+
+# Get current directory path
+CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
+
+from tools.web_search import DuckDuckGoSearchTool as WebSearch
+from tools.visit_webpage import VisitWebpageTool as VisitWebpage
+from tools.suggest_menu import SimpleTool as SuggestMenu
+from tools.catering_service_tool import SimpleTool as CateringServiceTool
+from tools.superhero_party_theme_generator import SuperheroPartyThemeTool as SuperheroPartyThemeGenerator
+from tools.final_answer import FinalAnswerTool as FinalAnswer
+
+
+
+model = HfApiModel(
+model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+provider=None,
+)
+
+web_search = WebSearch()
+visit_webpage = VisitWebpage()
+suggest_menu = SuggestMenu()
+catering_service_tool = CateringServiceTool()
+superhero_party_theme_generator = SuperheroPartyThemeGenerator()
+final_answer = FinalAnswer()
+
+
+with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
+
+agent = CodeAgent(
+    model=model,
+    tools=[web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator],
+    managed_agents=[],
+    max_steps=10,
+    verbosity_level=2,
+    grammar=None,
+    planning_interval=None,
+    name=None,
+    description=None,
+    prompt_templates=prompt_templates
+)
+if __name__ == "__main__":
+    GradioUI(agent).launch()