Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 from langgraph.graph import END, START, StateGraph
 
 from langchain_core.tracers.context import tracing_v2_enabled
@@ -18,7 +19,9 @@ from utils import (
 )
 
 
-def visualize_github_repo(repo_name, repo_branch):
+def visualize_github_repo(repo_name, repo_branch, mistral_api_key):
+    os.environ['MISTRAL_API_KEY'] = mistral_api_key
+    
     yield ("Looking at the Repo!", None)
     documents = load_github_codebase(repo_name, repo_branch)
     yield ("Repo loaded!", None)
@@ -105,6 +108,11 @@ demo = gr.Interface(
             value="main",  # Input component placeholder
             placeholder="Branch to explore",  # Input component description
         ),
+        gr.Textbox(
+            label="Mistral API Key",  # Input component label
+            # value="main",  # Input component placeholder
+            placeholder="Mistral API Key from mistral.ai",  # Input component description
+        ),
     ],
     outputs=[
         gr.Textbox(