github->hf

app.py

@@ -1,21 +1,13 @@
 import asyncio
+import json
 import os
 import threading
-import random
 from threading import Event
-from typing import Optional
-import json
 
 import discord
 import gradio as gr
-from discord import Permissions
 from discord.ext import commands
-from discord.utils import oauth_url
-
 from gradio_client import Client
-import gradio_client as grc
-from gradio_client.utils import QueueError
-
 
 event = Event()
 DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
@@ -23,8 +15,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 codellama_client = Client("https://huggingface-projects-codellama-13b-chat.hf.space/", HF_TOKEN)
 codellama_threadid_userid_dictionary = {}
 codellama_threadid_conversation = {}
-intents = discord.Intents.default()
-intents.message_content = True
+intents = discord.Intents.all()
 bot = commands.Bot(command_prefix="/", intents=intents)
 
 
@@ -42,11 +33,11 @@ async def on_ready():
     description="Enter a prompt to generate code!",
 )
 async def codellama(ctx, prompt: str):
-    """Audioldm2 generation"""
+    """Codellama generation"""
     try:
         await try_codellama(ctx, prompt)
     except Exception as e:
-        print(f"Error: (app.py){e}")
+        print(f"Error: {e}")
 
 
 @bot.event
@@ -60,7 +51,7 @@ async def on_message(message):
 
 
 async def try_codellama(ctx, prompt):
-    """Generates text based on a given prompt"""
+    """Generates code based on a given prompt"""
     try:
         global codellama_threadid_userid_dictionary
         global codellama_threadid_conversation
@@ -75,11 +66,11 @@ async def try_codellama(ctx, prompt):
         print(output_code)
         await thread.send(output_code)
     except Exception as e:
-        print(f"try_codellama Error: {e}")
+        print(f"Error: {e}")
 
 
 def codellama_initial_generation(prompt, thread):
-    """job.submit inside of run_in_executor = more consistent bot behavior"""
+    """Job.submit inside of run_in_executor = more consistent bot behavior"""
     global codellama_threadid_conversation
 
     chat_history = f"{thread.id}.json"
@@ -141,7 +132,7 @@ async def continue_codellama(message):
                             with open(chat_history, "w") as json_file:
                                 json.dump(conversation, json_file)
                             codellama_threadid_conversation[message.channel.id] = chat_history
-                            
+
                             if len(response) > 1300:
                                 response = response[:1300] + "...\nTruncating response due to discord api limits."
 
@@ -156,8 +147,9 @@ async def continue_codellama(message):
                                 await message.reply("Conversation ending due to length, feel free to start a new one!")
 
     except Exception as e:
-        print(f"continue_codellama Error: {e}")
-#---------------------------------------------------------------------------------------------------------------------
+        print(f"Error: {e}")
+
+
 def run_bot():
     if not DISCORD_TOKEN:
         print("DISCORD_TOKEN NOT SET")
@@ -170,11 +162,9 @@ threading.Thread(target=run_bot).start()
 event.wait()
 
 with gr.Blocks() as demo:
-    gr.Markdown(
-    """
+    gr.Markdown("""
     # Discord bot of https://huggingface.co/spaces/huggingface-projects/codellama-13b-chat
     https://discord.com/api/oauth2/authorize?client_id=1152238037355474964&permissions=326417516544&scope=bot
-    """
-    )
+    """)
 
-demo.launch()
+demo.launch()