File size: 1,103 Bytes
4962437
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
from discord.ext import commands
from langchain.llms import OpenAIChat
from swarms.agents import OmniModalAgent

# Setup
TOKEN = 'YOUR_DISCORD_BOT_TOKEN'
bot = commands.Bot(command_prefix='!')

# Initialize the OmniModalAgent
llm = OpenAIChat(model_name="gpt-4")
agent = OmniModalAgent(llm)

@bot.event
async def on_ready():
    print(f'We have logged in as {bot.user}')

@bot.command()
async def greet(ctx):
    """Greets the user."""
    await ctx.send(f'Hello, {ctx.author.name}!')

@bot.command()
async def run(ctx, *, description: str):
    """Generates a video based on the given description."""
    response = agent.run(description)  # Assuming the response provides information or a link to the generated video
    await ctx.send(response)

@bot.command()
async def help_me(ctx):
    """Provides a list of commands and their descriptions."""
    help_text = """
    - `!greet`: Greets you.
    - `!run [description]`: Generates a video based on the given description.
    - `!help_me`: Provides this list of commands and their descriptions.
    """
    await ctx.send(help_text)

bot.run(TOKEN)