Update app.py

app.py

@@ -3,50 +3,66 @@ import datetime
 import requests
 import pytz
 import yaml
+import io
+import base64
+import schemdraw
+import schemdraw.elements as elm
 from tools.final_answer import FinalAnswerTool
-
 from Gradio_UI import GradioUI
 
 @tool
-def simulate_ai_debate(topic: str, rounds: int = 3) -> str:
+def text_to_flowchart(steps_text: str) -> str:
     """
-    A tool that simulates a debate between two AI models (AI Alpha and AI Beta) on a given topic.
+    Generates a flowchart diagram from pre-processed text that lists sequential process steps.
+    The input should be a text with each step on a new line (optionally prefixed by bullet markers).
     
     Args:
-        topic: The subject of the discussion.
-        rounds: Number of exchange rounds between the two AI models.
+        steps_text: A string containing the process steps.
     
     Returns:
-        A transcript of the simulated debate.
+        A data URL for a PNG image of the generated flowchart.
     """
-    transcript = f"Debate Topic: {topic}\n\n"
+    # Parse steps from the input text
+    parsed_steps = []
+    for line in steps_text.splitlines():
+        line = line.strip()
+        if line.startswith(("-", "*", "•")):
+            line = line[1:].strip()
+        elif line and line[0].isdigit():
+            dot_index = line.find('.')
+            if dot_index != -1:
+                line = line[dot_index+1:].strip()
+        if line:
+            parsed_steps.append(line)
+    if not parsed_steps:
+        parsed_steps = ["No steps provided."]
+    
+    # Create the flowchart using SchemDraw
+    d = schemdraw.Drawing()
+    d.add(elm.Start(label="Start"))
+    for step in parsed_steps:
+        d.add(elm.Arrow())
+        d.add(elm.Process().label(step))
+    d.add(elm.Arrow())
+    d.add(elm.End(label="End"))
     
-    for i in range(rounds):
-        # AI Alpha's turn
-        alpha_prompt = (
-            f"Round {i+1} - You are AI Alpha. Present your argument on the topic '{topic}' "
-            f"given the following discussion transcript:\n\n{transcript}\nYour response:"
-        )
-        alpha_response = model.generate(prompt=alpha_prompt)
-        transcript += f"AI Alpha: {alpha_response}\n\n"
-        
-        # AI Beta's turn
-        beta_prompt = (
-            f"Round {i+1} - You are AI Beta. Provide your counterargument on the topic '{topic}' "
-            f"given the following discussion transcript:\n\n{transcript}\nYour response:"
-        )
-        beta_response = model.generate(prompt=beta_prompt)
-        transcript += f"AI Beta: {beta_response}\n\n"
+    # Render diagram to PNG and encode as a data URL
+    buf = io.BytesIO()
+    d.draw()
+    d.fig.savefig(buf, format='png', bbox_inches='tight')
+    buf.seek(0)
+    encoded_image = base64.b64encode(buf.getvalue()).decode('utf-8')
+    data_url = f"data:image/png;base64,{encoded_image}"
     
-    return transcript
+    return data_url
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """
-    A tool that fetches the current local time in a specified timezone.
+    Fetches the current local time for a specified timezone.
     
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+        timezone: A valid timezone string (e.g., 'America/New_York').
     
     Returns:
         A string with the current local time or an error message.
@@ -71,16 +87,16 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
-    
+
 agent = CodeAgent(
     model=model,
-    tools=[simulate_ai_debate, get_current_time_in_timezone, final_answer],
+    tools=[text_to_flowchart, final_answer],
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
-    name=None,
-    description=None,
+    name="Flowchart Generator",
+    description="Generates a flowchart diagram from pre-processed sequential steps using SchemDraw.",
     prompt_templates=prompt_templates
 )