add persistence

app.py

@@ -3,6 +3,12 @@ import discord
 from gradio_client import Client
 from huggingface_hub import InferenceClient
 import os
+import logging
+import gradio as gr
+import threading
+
+# Set up logging
+logging.basicConfig(level=logging.INFO, format='[%(asctime)s] [%(levelname)s] %(message)s')
 
 # Get tokens from environment variables (Hugging Face Spaces secrets)
 TOKEN = os.getenv("DISCORD_TOKEN")
@@ -25,11 +31,7 @@ hf_client = InferenceClient(api_key=HF_TOKEN)
 # Function to process message and get response
 async def get_ai_response(message_content):
     try:
-        messages = [
-            {"role": "user", "content": message_content}
-        ]
-        
-        # Create a streaming response
+        messages = [{"role": "user", "content": message_content}]
         response = ""
         stream = hf_client.chat.completions.create(
             model="Qwen/Qwen2.5-72B-Instruct",
@@ -39,44 +41,29 @@ async def get_ai_response(message_content):
             top_p=0.7,
             stream=True
         )
-        
-        # Collect the streamed response
         for chunk in stream:
             content = chunk.choices[0].delta.content
             if content:
                 response += content
-                
         return response if response else "I couldn't generate a response."
-        
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
 @client.event
 async def on_ready():
-    print(f'We have logged in as {client.user}')
+    logging.info(f'We have logged in as {client.user}')
 
 @client.event
 async def on_message(message):
-    # Ignore messages from the bot itself
     if message.author == client.user:
         return
-
-    # Check if the bot is mentioned in the message
     if client.user in message.mentions:
-        # Extract the message content without the bot mention
         clean_message = message.content.replace(f"<@{client.user.id}>", "").strip()
-        
         if not clean_message:
             await message.channel.send("Please provide some text for me to respond to!")
             return
-            
-        # Send initial response to show bot is processing
         processing_message = await message.channel.send("Processing your request...")
-        
-        # Get AI response
         response = await get_ai_response(clean_message)
-        
-        # Split response if it's too long for Discord's 2000 character limit
         if len(response) > 2000:
             chunks = [response[i:i+2000] for i in range(0, len(response), 2000)]
             await processing_message.delete()
@@ -85,21 +72,34 @@ async def on_message(message):
         else:
             await processing_message.edit(content=response)
 
-# Error handling for connection issues
 @client.event
 async def on_error(event, *args, **kwargs):
-    print(f"An error occurred: {event}")
+    logging.error(f"An error occurred: {event}")
     with open('error.log', 'a') as f:
         f.write(f"{event}\n")
 
-# Run the bot
-def main():
+# Function to run the Discord bot in a separate thread
+def run_discord_bot():
     try:
+        logging.info("Starting the Discord bot...")
         client.run(TOKEN)
     except Exception as e:
-        print(f"Failed to start bot: {e}")
+        logging.error(f"Failed to start bot: {e}")
         with open('error.log', 'a') as f:
             f.write(f"Failed to start bot: {e}\n")
 
+# Gradio interface to keep the Space alive
+def create_interface():
+    invite_url = "Add this bot to your server by following this URL: https://discord.com/oauth2/authorize?client_id=1347942347077582880&permissions=377957238784&integration_type=0&scope=bot"
+    with gr.Blocks(title="Discord Bot Invite") as demo:
+        gr.Markdown(f"# Discord Bot\n{invite_url}")
+    return demo
+
 if __name__ == "__main__":
-    main()
+    # Start the Discord bot in a separate thread
+    bot_thread = threading.Thread(target=run_discord_bot, daemon=True)
+    bot_thread.start()
+
+    # Launch the Gradio interface
+    interface = create_interface()
+    interface.launch(server_name="0.0.0.0", server_port=7860)