Update app.py

app.py

@@ -53,6 +53,40 @@ async def uptime(ctx):
     embed.set_footer(text="Created by Cosmos")
 
     await ctx.send(embed=embed)
+    
+session_count = 0
+
+@bot.command()
+async def ai(ctx, *, input_text: str):
+    """Ask our AI model a question. (Session resets every 10 messages!)"""
+    global session_count  # Access the global variable
+    
+    # Increment session count
+    session_count += 1
+    
+    try:
+        # Check if it's time to start a new session
+        if session_count % 10 == 0:
+            # Start a new session by passing a unique session hash
+            client = get_client(session=str(datetime.datetime.now()))
+        else:
+            # Use the existing session
+            client = get_client()
+        
+        # Make a request to the /chat endpoint
+        result = client.predict(
+            input_text,
+            0.9,  # Temperature value between 0.0 and 1.0
+            2000,  # Max new tokens value between 64 and 4096
+            0.9,  # Top-p (nucleus sampling) value between 0.0 and 1.0
+            1.2,  # Repetition penalty value between 1.0 and 2.0
+            api_name="/chat"
+        )
+        
+        # Send the AI's response
+        await ctx.send(result)
+    except Exception as e:
+        await ctx.send(f"An error occurred: {str(e)}")
 
 @bot.command()
 async def verse(ctx):