Update app.py

app.py

@@ -5,15 +5,11 @@ from datetime import datetime
 from typing import Literal
 import os
 import importlib
-from llm_handler import send_to_llm
+from llm_handler import send_to_llm, agent, settings
 from main import generate_data, PROMPT_1
 from topics import TOPICS
 from system_messages import SYSTEM_MESSAGES_VODALUS
 import random
-import llamanet
-
-# Initialize LlamaNet
-llamanet.run()
 
 ANNOTATION_CONFIG_FILE = "annotation_config.json"
 OUTPUT_FILE_PATH = "dataset.jsonl"
@@ -276,7 +272,7 @@ def chat_with_llm(message, history):
             msg_list.append({"role": "assistant", "content": h[1]})
         msg_list.append({"role": "user", "content": message})
 
-        response, _ = send_to_llm("llamanet", msg_list)
+        response, _ = send_to_llm(agent, msg_list)
         
         return history + [[message, response]]
     except Exception as e: