Update app.py

app.py

@@ -6,7 +6,16 @@ import subprocess
 import threading
 import time
 import shutil
-from typing import Dict, Tuple
+from typing import Dict, Tuple, List
+import json
+from rich import print as rprint
+from rich.panel import Panel
+from rich.progress import track
+from rich.table import Table
+from rich.prompt import Prompt, Confirm
+from rich.markdown import Markdown
+from rich.traceback import install
+install()  # Enable rich tracebacks for easier debugging
 
 # --- Constants ---
 
@@ -312,8 +321,8 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
         submit_button.click(
             chat_interface,
             inputs=[
-                chat_interface_input._id,
-                chatbot._id,
+                chat_interface_input,
+                chatbot,
                 get_agent_cluster,
                 temperature_slider,
                 max_new_tokens_slider,
@@ -370,5 +379,8 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
         outputs=[code_output],
     )
 
+    # --- Initialize Hugging Face Client ---
+    client = InferenceClient(repo_id=MODEL_NAME, token=os.environ.get("HF_TOKEN"))
+
     # --- Launch Gradio ---
     demo.queue().launch(debug=True)