Update app.py

app.py

@@ -9,7 +9,6 @@ import gradio as gr
 from discord import Permissions
 from discord.ext import commands
 from discord.utils import oauth_url
-from discord.ext import Option
 import gradio_client as grc
 from gradio_client.utils import QueueError
 
@@ -35,10 +34,11 @@ def truncate_response(response: str) -> str:
     else:
         return response
 
+thread_to_client = {}
+thread_to_user = {}
 intents = discord.Intents.default()
 intents.message_content = True
-bot = commands.Bot(command_prefix="/", intents=intents)
-
+bot = commands.Bot(command_prefix="!", intents=intents)
 
 
 ## BOT COMMANDS ##
@@ -56,41 +56,37 @@ async def on_ready():
     url = oauth_url(bot.user.id, permissions=permissions)
     print(f"Add this bot to your server by clicking this link: {url}")
 
-thread_to_client = {}
-thread_to_user = {}
 
-## /echo command to return the llm repsonse ##
-@bot.hybrid_command(name="cryptoportfolio", description="Enter text and parameters like this: /cryptoportfolio ")
-async def chat(ctx, prompt: str):
-    if ctx.author.id == bot.user.id:
-        return
-@bot.hybrid_command(name="cryptoportfolio", description="Select a predefined sentence")
-async def crypto_portfolio(ctx, 
-                           selection: Option(str, 
-                                             "Select an option", 
-                                             choices=[
-                                                 "Find the top performing cryptocurrencies of the day.",
-                                                 "Find the top performing cryptocurrencies of the week.",
-                                                 "Find the top performing cryptocurrencies of the month."
-                                             ])):
-    try:
-        # Acknowledge the command
-        await ctx.send(f"Processing request: {selection}")
-
-        # Simulated external processing
-        loop = asyncio.get_running_loop()
-        client = await loop.run_in_executor(None, get_client, None)
-        job = client.submit(selection, api_name="/predict")
-        await wait(job)
+@bot.command(name="cryptoportfolio", help="Select a predefined sentence by typing it")
+async def crypto_portfolio(ctx, *, input_text: str):
+    valid_choices = [
+        "Find the top performing cryptocurrencies of the day.",
+        "Find the top performing cryptocurrencies of the week.",
+        "Find the top performing cryptocurrencies of the month."
+    ]
 
+    if input_text in valid_choices:
         try:
-            job.result()
-            response = job.outputs()[-1]
-            await ctx.send(f"```{truncate_response(response)}```")  # Send the LLM response
-        except QueueError:
-            await ctx.send("The gradio space powering this bot is really busy! Please try again later!")
-    except Exception as e:
-        print(f"{e}")
+            # Acknowledge the command
+            await ctx.send(f"Processing request: {input_text}")
+
+            # Simulated external processing (replace this with your actual logic)
+            loop = asyncio.get_running_loop()
+            client = await loop.run_in_executor(None, get_client, None)
+            job = client.submit(input_text, api_name="/predict")
+            await wait(job)
+
+            try:
+                job.result()
+                response = job.outputs()[-1]
+                await ctx.send(f"```{truncate_response(response)}```")  # Send the LLM response
+            except QueueError:
+                await ctx.send("The gradio space powering this bot is really busy! Please try again later!")
+        except Exception as e:
+            print(f"{e}")
+    else:
+        # Handle invalid input
+        await ctx.send("Invalid input. Please select one of the predefined sentences.")
 
 
 @bot.hybrid_command(name="cshelp", description="Get information on how to use this bot.")