Spaces:
Running
Running
File size: 2,467 Bytes
e0fa04b bb04558 5bd4d70 e0fa04b bb04558 e0fa04b bb04558 778ded8 bb04558 5bd4d70 bb04558 5bd4d70 bb04558 f51f1c6 e6440a0 f51f1c6 5bd4d70 c57b527 a063c0b e6440a0 5bd4d70 d7eef58 e0fa04b |
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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
import discord
from discord import app_commands
import os
import requests
import asyncio
import aiohttp # Use aiohttp for asynchronous HTTP requests
import gradio as gr # Import Gradio
import google.generativeai as genai
from io import BytesIO
import PIL.Image
# --- Environment Variables & Setup ---
DISCORD_BOT_TOKEN = os.getenv("DISCORD_BOT_TOKEN")
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
# --- Discord Bot Setup ---
intents = discord.Intents.default()
client = discord.Client(intents=intents)
tree = app_commands.CommandTree(client)
if not DISCORD_BOT_TOKEN or not GEMINI_API_KEY:
raise ValueError("Both DISCORD_BOT_TOKEN and GEMINI_API_KEY must be set.")
@tree.command(name="hello", description="Says hello!")
async def hello_command(interaction):
await interaction.response.send_message("Hello there!")
@tree.command(name="gemini", description="chat with gemini")
# @app_commands.choices(
# aspect_ratio=[
# app_commands.Choice(name="1:1 (Square)", value="1:1"),
# app_commands.Choice(name="9:16 (Vertical)", value="9:16"),
# app_commands.Choice(name="16:9 (Horizontal)", value="16:9"),
# app_commands.Choice(name="3:4", value="3:4"),
# app_commands.Choice(name="4:3", value="4:3"),
# ]
# )
async def generate_command(
interaction: discord.Interaction,
prompt: str,
# img_count: int,
# aspect_ratio: app_commands.Choice[str],
):
try:
await interaction.response.defer() # Defer the interaction
genai.configure(api_key=GEMINI_API_KEY)
model = genai.GenerativeModel("gemini-2.0-flash-exp")
response = model.generate_content(prompt, stream=True)
for part in response:
await interaction.followup.send(content=part.text)
except Exception as e:
print(e)
async def on_ready():
await tree.sync()
print("Bot is ready!")
client.event(on_ready)
# --- Gradio Interface ---
def echo_text(text):
return text
def run_gradio():
gr.Interface(
fn=echo_text,
inputs="text",
outputs="text",
live=False,
title="Minimal Gradio Interface",
).launch(server_name="0.0.0.0", server_port=7860, share=False, show_error=True)
async def main():
bot_task = asyncio.create_task(client.start(DISCORD_BOT_TOKEN))
gradio_task = asyncio.to_thread(run_gradio)
await asyncio.gather(bot_task, gradio_task)
if __name__ == "__main__":
asyncio.run(main())
|