Update app.py

app.py

@@ -72,25 +72,17 @@ def visualize_github_repo(repo_name, repo_branch, mistral_api_key):
                 print(s)
                 print("----")
                 if "planner" in s:
-                    yield ("Planning the Exploration !")#, None)
+                    yield "Planning the Exploration !"#, None)
                 if "parse_plan" in s:
-                    yield ("Planning done! Parsing the plan!")#, None)
+                    yield "Planning done! Parsing the plan!"#, None)
                 if "explore_file" in s:
-                    yield (
-                        f"Exploration started! Exploring file: {s['explore_file']['explored_files'][-1]} !",
-                        # None,
-                    )
+                    yield f"Exploration started! Exploring file: {s['explore_file']['explored_files'][-1]} !",
+
                 if "generate_mermaid_code" in s:
-                    yield (
-                        "Exploration done! Gathering thoughts and generating a graph!",
-                        # None,
-                    )
+                    yield "Exploration done! Gathering thoughts and generating a graph!",
                 if "render_mermaid" in s:
 
-                    yield (
-                        s["render_mermaid"]["messages"][-1].content,
-                        # s["render_mermaid"]["final_graph"],
-                    )
+                    yield s["render_mermaid"]["messages"][-1].content
 
 
 demo = gr.Interface(