Update app.py

app.py

@@ -1,26 +1,27 @@
 import os
-from langgraph.graph import END, START, StateGraph
 
-from langchain_core.tracers.context import tracing_v2_enabled
-
-from prompts import *
-import gradio as gr
-
-
-from utils import (
-    load_github_codebase,
-    router,
-    get_plan_for_codebase,
-    parse_plan,
-    explore_file,
-    final_mermaid_code_generation,
-    extract_mermaid_and_generate_graph,
-    GraphState,
-)
 
 
 def visualize_github_repo(repo_name, repo_branch, mistral_api_key):
     os.environ['MISTRAL_API_KEY'] = mistral_api_key
+
+    from langgraph.graph import END, START, StateGraph
+    from langchain_core.tracers.context import tracing_v2_enabled
+    
+    from prompts import *
+    import gradio as gr
+    
+    
+    from utils import (
+        load_github_codebase,
+        router,
+        get_plan_for_codebase,
+        parse_plan,
+        explore_file,
+        final_mermaid_code_generation,
+        extract_mermaid_and_generate_graph,
+        GraphState,
+    )
     
     yield ("Looking at the Repo!", None)
     documents = load_github_codebase(repo_name, repo_branch)