Upload PromptBot.py

PromptBot.py

@@ -0,0 +1,64 @@
+import discord
+import requests
+import json
+import os
+import asyncio
+
+# Create an instance of Intents
+intents = discord.Intents.default()
+intents.message_content = True  # Enable message content events
+
+client = discord.Client(intents=intents)
+
+@client.event
+async def on_ready():
+    print(f'We have logged in as {client.user}')
+
+@client.event
+async def on_message(message):
+    if message.author == client.user:
+        return
+
+    # Change the command trigger to "/prompt"
+    if message.content.startswith('/prompt'):
+        input_text = message.content[len('/prompt'):].strip()
+        if input_text:
+            # Mention the user in the response
+            user_mention = message.author.mention
+
+            print(f'Received message: {message.content}')
+            
+            # Send typing indicator
+            print(f'Sending typing indicator...')
+            async with message.channel.typing():
+                print(f'Typing indicator sent.')
+                
+                # Simulate a delay (you can adjust this based on the actual response time)
+                print(f'Simulating delay...')
+                await asyncio.sleep(2)
+
+                # Generate the prompt
+                print(f'Generating prompt...')
+                prompt = get_prompt(input_text)
+
+                # Reply directly to the input message
+                print(f'Replying to message...')
+                await message.reply(f'{user_mention} Generated Prompt: {prompt}')
+                print(f'Reply sent.')
+
+def get_prompt(input_text):
+    api_url = "https://gustavosta-magicprompt-stable-diffusion.hf.space/api/predict"
+    headers = {"Content-Type": "application/json"}
+    payload = {"data": [input_text]}
+
+    print(f'Sending API request...')
+    response = requests.post(api_url, data=json.dumps(payload), headers=headers)
+    result = response.json()
+
+    print(f'API response: {result}')
+
+    return result['data'][0] if result['data'][0] else 'Prompt generation failed.'
+
+# Replace 'YOUR_BOT_TOKEN' with the actual token you obtained from the Discord Developer Portal
+print('Starting the bot...')
+client.run('MTE2OTIzNDc0ODc1Mzg0NjMxMg.GOlz8M.rQlvjfQWLYPNpEhWVQ0vJ7sUw_IV0TooMdN8Os')