From 2b9b2ad77857add2fa864f7c05665e72c18445bf Mon Sep 17 00:00:00 2001
From: nex <me@nexy7574.co.uk>
Date: Sat, 18 Nov 2023 19:17:38 +0000
Subject: [PATCH] Set faststart flag to all M4A files

---
 assets/china.m4a        | Bin 89951 -> 89951 bytes
 assets/common-sense.m4a | Bin 41087 -> 41087 bytes
 assets/disgrace.m4a     | Bin 80657 -> 80657 bytes
 assets/drones.m4a       | Bin 71777 -> 71777 bytes
 assets/hair.m4a         | Bin 240827 -> 240827 bytes
 assets/peppa-pig.m4a    | Bin 269126 -> 269126 bytes
 assets/pork.m4a         | Bin 54367 -> 54367 bytes
 assets/scrapped.m4a     | Bin 217762 -> 217762 bytes
 assets/tea.m4a          | Bin 125308 -> 125308 bytes
 cogs/other.py           | 150 ++++++++++++++++++++++++----------------
 10 files changed, 92 insertions(+), 58 deletions(-)

diff --git a/assets/china.m4a b/assets/china.m4a
index 4fc444765124b0c0a68722265d65b8edc5da99b9..6cff7141c092a2b4ae6e9285c332e343e94024d5 100644
GIT binary patch
delta 29
lcmcb=kM;gO)(LXk`9m4Eb24$&O+OLGIK5G3s|@4E5CF8W3t|8O

delta 28
kcmcb=kM;gO)(LXW1zg(;xESAuY?Rr_$;6;D`8wxx0I=!{EC2ui

diff --git a/assets/common-sense.m4a b/assets/common-sense.m4a
index ae571a61f8b4a8138f79079dd0f004656beea610..dce437a1b46b083f58c3364b86326ec90d33c282 100644
GIT binary patch
delta 24
gcmex=fa(7MrU`PJ&1bD=WnwFv{9^v}jWUw60F#Id7ytkO

delta 24
gcmex=fa(7MrU`PJGuXsuZIs!_%EX{E`8?}%0EqGlvH$=8

diff --git a/assets/disgrace.m4a b/assets/disgrace.m4a
index 51143e3d7be01d935c282b2c4ec4ea883fb91f6b..f4a7d569e6f449b25b285bdc0808b2a6727f531b 100644
GIT binary patch
delta 29
lcmbREjb-9DmI-p(YYiEHb1-opnqFnaIK5G3s|;hjApo}!3f%wz

delta 28
kcmbREjb-9DmI-prTR69G;be?8+$i&xgNZ?BvNY#(0IUECRsaA1

diff --git a/assets/drones.m4a b/assets/drones.m4a
index bc956d22623177bb691e44f9182804891154d3f4..477edade1d510dcdc7346c5af25cc8462c2bc5b3 100644
GIT binary patch
delta 29
lcmaF3f#u-_mI-p(C-O6zb1-rIn|?~1aeAZ7RvE@(egLrE3V8qk

delta 28
kcmaF3f#u-_mI-prA2_yu;9xA|-za0r!Nj05Ih<oU0I%x`NdN!<

diff --git a/assets/hair.m4a b/assets/hair.m4a
index e5f072c171dd8f24bae3473136b008a2826ff957..b58542a3a757b6c45631a5ef0b7b5478ac05d690 100644
GIT binary patch
delta 35
tcmV+;0Nnq(*$%te4v-$V_mu-H76KOXmn5tMmw_CG907$K0<|0i*_94H4XFSC

delta 36
rcmdmelW+G;z6o;8PsG}vh%o{&)AlD~%-5%Gl+hJqV$hlFEjAqhEU^w=

diff --git a/assets/peppa-pig.m4a b/assets/peppa-pig.m4a
index a9c319fd763d8e289216e76b3b7c9ddc94aeb5a6..4a35c02afca71c1f47d96ffde2fced291db6ef27 100644
GIT binary patch
delta 38
ucmX@MP2kuzfeCWk{g_zdC72}DrmvS`ncgVVD#O?+!_+Fnyj6z9fC&KpsS6SS

delta 40
tcmX@MP2kuzfeCWW5|Zr_l8ivi1jNkSB_vt&m^R8JN-#0#Os<!h4gmaY3i<#5

diff --git a/assets/pork.m4a b/assets/pork.m4a
index 41fc21545d36149e594e68309b068fc4e806d56d..d64e2659d33dcf66cfcbdabd4baaea33b3171beb 100644
GIT binary patch
delta 24
icmV+z0O$YTr~}`q1CSoG`^?J*0tb?l$kUgx9No;E(G27O

delta 24
gcmcb=g8BXm<_U6}Ti9=(*(h_Cjfp{L@@ux~0FKKFkpKVy

diff --git a/assets/scrapped.m4a b/assets/scrapped.m4a
index 44a96a66e10581251e09cc2926faaa5ea28e7790..b8680bcff2744358fd9a18a6e6e15d5af647fa0f 100644
GIT binary patch
delta 35
rcmZ3qly}im-U)KsrR<pB3p0sqo6a4?JiSq-Rfe%uhH0w|^B-FP;G+w)

delta 36
rcmZ3qly}im-U)KeD@58?h%f>%)Aki2%)e|m%6t@NV$hk)FESkf2FMK<

diff --git a/assets/tea.m4a b/assets/tea.m4a
index f9301d67020fd69c9846069cfa778191e23a1d1a..f45d51d6715298f7770f7e1c0714bbde1e8df72d 100644
GIT binary patch
delta 29
lcmex!iT%$d_6c&^ZSF8G=V9W_oBs3#<Mc+Etul;qcL31+488yW

delta 28
kcmex!iT%$d_6c&$3B21Aco}8xY?N8a!^EI7`8dyX0JMV(6951J

diff --git a/cogs/other.py b/cogs/other.py
index 3442f8e..4b8a861 100644
--- a/cogs/other.py
+++ b/cogs/other.py
@@ -1,5 +1,6 @@
 import asyncio
 import base64
+import fnmatch
 import functools
 import glob
 import io
@@ -1897,70 +1898,103 @@ class OtherCog(commands.Cog):
                 return await ctx.respond(":x: Context not found in cache.")
             context = self.context_cache[context]
 
-        content = None
-        RESTRICTED_SERVERS = (
-            "100.106.34.86:11434",
-            "100.66.187.46:11434"
-        )
-        try_hosts = {
-            "100.106.34.86:11434": "NexTop",
-            "ollama.shronk.net": "Alibaba Cloud",
-            "100.66.187.46:11434": "NexBox",
-        }
         model = model.casefold()
+        try:
+            model.split(":", 1)
+        except ValueError:
+            model = model + ":latest"
 
-        if server == "auto":
-            async with httpx.AsyncClient(follow_redirects=True) as client:
-                for host in try_hosts.keys():
-                    try:
-                        response = await client.get(
-                            f"http://{host}/api/tags",
-                        )
-                        response.raise_for_status()
-                    except (httpx.TransportError, httpx.NetworkError, httpx.HTTPStatusError):
-                        continue
-                    else:
-                        break
-                else:
-                    return await ctx.respond(":x: No servers available.")
-        else:
-            try:
-                try:
-                    server, port = server.split(":", 1)
-                    port = int(port)
-                except ValueError:
-                    port = 11434
-                server = ipaddress.ip_address(server)
-                if not isinstance(server, ipaddress.IPv4Address):
-                    raise ValueError
-                server = "%s:%s" % (server, port)
-            except ValueError:
-                try:
-                    if not server.startswith("http"):
-                        server = "http://" + server
-                    server = urlparse(server)
-                    if not server.netloc:
-                        raise ValueError
-                except ValueError:
-                    return await ctx.respond(f":x: Failed to parse {server!r} as a domain/IPv4.")
-                else:
-                    _server = server
-                    server = _server.netloc
-                    port = _server.port or 443
-                    server = "%s:%s" % (server, port)
+        servers = {
+            "100.106.34.86:11434": {
+                "name": "NexTop",
+                "allow": [
+                    "orca-mini:latest",
+                    "orca-mini:3b",
+                    "orca-mini:7b",
+                    "llama2:latest",
+                    "llama2-uncensored:latest",
+                    "codellama:latest",
+                    "codellama:python",
+                    "codellama:instruct",
+                ],
+                "owner": 421698654189912064
+            },
+            "ollama.shronk.net:11434": {
+                "name": "Alibaba Cloud",
+                "allow": ["*"],
+                "owner": 1019233057519177778,
+            },
+            "100.66.187.46:11434": {
+                "name": "NexBox",
+                "restrict-to": [
+                    "orca-mini:latest",
+                    "orca-mini:3b",
+                    "orca-mini:7b",
+                ],
+                "owner": 421698654189912064
+            },
+        }
 
-            host = server
+        class ServerSelector(discord.ui.View):
+            def __init__(self):
+                super().__init__(
+                    disable_on_timeout=True
+                )
+                self.chosen_server = None
 
-        if host in RESTRICTED_SERVERS:
-            if not await self.bot.is_owner(ctx.author):
-                if not model.startswith("orca-mini"):
-                    await ctx.respond(
-                        ":warning: You can only use `orca-mini` models.",
-                        delete_after=30,
-                        ephemeral=True
+            @discord.ui.select(
+                placeholder="Choose a server.",
+                custom_id="select",
+                options=[
+                    discord.SelectOption(
+                        label="%s (%s)" % (y['name'], x),
+                        label=x
                     )
-                    model = "orca-mini"
+                    for x, y in servers.items()
+                    if fnmatch.fnmatch(model, y.get("restrict-to", ['*']))
+                ] + [
+                    discord.SelectOption(
+                        label="Custom",
+                        value="custom"
+                    )
+                ]
+            )
+            async def select_callback(self, item: discord.ui.Select, interaction: discord.Interaction):
+                if item.values[0] == "custom":
+                    class ServerSelectionModal(discord.ui.Modal):
+                        def __init__(self):
+                            super().__init__(
+                                discord.ui.InputText(
+                                    label="Domain/IP",
+                                    placeholder="Enter server",
+                                    min_length=1,
+                                    max_length=4000,
+                                    style=discord.InputTextStyle.short,
+                                ),
+                                discord.ui.InputText(
+                                    label="Port (only 443 is HTTPS)",
+                                    placeholder="11434",
+                                    min_length=2,
+                                    max_length=5,
+                                    style=discord.InputTextStyle.short,
+                                    value="11434"
+                                ),
+                                title="Enter server details",
+                                timeout=120
+                            )
+                            self.hostname = None
+                            self.port = None
 
+                        async def callback(self, _interaction: discord.Interaction):
+                            await _interaction.response.defer()
+                            self.stop()
+
+                    _modal = ServerSelectionModal()
+                    await interaction.response.send_modal(_modal)
+                    await _modal.wait()
+                    if not all()
+
+        content = None
         embed = discord.Embed(
             colour=discord.Colour.greyple()
         )