Update app.py

app.py

@@ -6,7 +6,7 @@ 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_wrapper
 from main import generate_data, PROMPT_1
 from topics import TOPICS
 from system_messages import SYSTEM_MESSAGES_VODALUS
@@ -469,12 +469,8 @@ 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(msg_list)
-        
-        return history + [[message, response]]
-    except Exception as e:
-        print(f"Error in chat_with_llm: {str(e)}")
-        return history + [[message, f"Error: {str(e)}"]]
+        # Update this line to use send_to_llm_wrapper
+        response, _ = send_to_llm_wrapper(msg_list)
         
         return history + [[message, response]]
     except Exception as e: