Update app.py

app.py

@@ -144,8 +144,8 @@ client = discord.Client(intents=intents)
 async def handle_teapot_inference(server_name, user_input):
     teapot_instance = CONFIG.get(server_name, CONFIG["Teapot AI"])
     print(f"Using Teapot instance for server: {server_name}")
-    response, debug_info = await teapot_instance.infer(user_input)
-    return response, debug_info
+    response = await teapot_instance.query(user_input)
+    return response
 
 
 @client.event
@@ -165,22 +165,17 @@ async def on_message(message):
     server_name ="OneTrainer"
     async with message.channel.typing():
         cleaned_message = message.content.replace(f'<@{client.user.id}>', "").strip()
-        response, debug_info = await handle_teapot_inference(server_name, cleaned_message)
+        response = await handle_teapot_inference(server_name, cleaned_message)
         sent_message = await message.reply(response)
 
-        if is_debug:
-            thread = await sent_message.create_thread(
-                name=f"Debug Thread: '{cleaned_message[:80]}'",
-                auto_archive_duration=60
-            )
-            await thread.send(debug_info)
 
 # ========= STREAMLIT =========
 @st.cache_resource
-def initialize():
+def discord_loop():
     st.session_state["initialized"] = True
     client.run(DISCORD_TOKEN)
+    st.write("418 I'm a teapot")
     return
 
-st.write("418 I'm a teapot")
-initialize()
+
+discord_loop()