Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, tool, load_tool
 import datetime
 import requests
 import pytz
@@ -6,7 +6,7 @@ import yaml
 import io
 import base64
 import schemdraw
-import schemdraw.elements as elm
+from schemdraw import flow
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
@@ -37,14 +37,14 @@ def text_to_flowchart(steps_text: str) -> str:
     if not parsed_steps:
         parsed_steps = ["No steps provided."]
     
-    # Create the flowchart using SchemDraw
-    d = schemdraw.Drawing()
-    d.add(elm.Roundbox().label("Start"))
+    # Create the flowchart using SchemDraw's flow module
+    d = schemdraw.Drawing(unit=0.5, fontsize=10)
+    d += flow.Terminal().label("Start")
     for step in parsed_steps:
-        d.add(elm.Arrow())
-        d.add(elm.Box().label(step))
-    d.add(elm.Arrow())
-    d.add(elm.Roundbox().label("End"))
+        d += flow.Arrow()
+        d += flow.Process().label(step)
+    d += flow.Arrow()
+    d += flow.Terminal().label("End")
     
     # Render diagram to PNG and encode as a data URL
     buf = io.BytesIO()
@@ -76,18 +76,16 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 final_answer = FinalAnswerTool()
 
-model = HfApiModel(
+model = load_tool("HfApiModel", trust_remote_code=True)(
     max_tokens=2096,
     temperature=0.5,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
     custom_role_conversions=None,
 )
 
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
-
+    
 agent = CodeAgent(
     model=model,
     tools=[text_to_flowchart, final_answer],
@@ -96,7 +94,7 @@ agent = CodeAgent(
     grammar=None,
     planning_interval=None,
     name="Flowchart Generator",
-    description="Generates a flowchart diagram from pre-processed sequential steps using SchemDraw.",
+    description="Generates a flowchart diagram from sequential steps using SchemDraw's flow module.",
     prompt_templates=prompt_templates
 )