Initial commit

app.py

@@ -48,17 +48,18 @@ class CustomDNSResolver:
             logger.error(f"DNS resolution error for {host}: {e}")
             return None
 
-class CustomConnectionPool(AsyncConnectionPool):
-    def __init__(self, custom_resolver, *args, **kwargs):
-        self.custom_resolver = custom_resolver
-        super().__init__(*args, **kwargs)
-
-    async def create_connection(self, origin, timeout=None, socket_options=None):
-        host, port = origin
-        addresses = await self.custom_resolver.resolve(host, port)
-        if not addresses:
-            raise Exception(f"Could not resolve {host}")
-        return await super().create_connection((addresses[0][0], port), timeout, socket_options)
+class CustomDNSResolver:
+    def __init__(self):
+        self.resolver = aiodns.DNSResolver(nameservers=['8.8.8.8', '8.8.4.4'])
+
+    async def resolve(self, hostname):
+        try:
+            result = await self.resolver.query(hostname, 'A')
+            return [{'host': entry.host, 'port': 443} for entry in result]
+        except aiodns.error.DNSError as e:
+            logger.error(f"DNS resolution error for {hostname}: {e}")
+            return None
+
 
 class TranscribeKHBot:
     def __init__(self):
@@ -70,11 +71,12 @@ class TranscribeKHBot:
         self.initialize_mms_model()
         self.initialize_whisper_model()
         
-        self.custom_pool = CustomConnectionPool(self.custom_resolver)
         self.http_client = httpx.AsyncClient(
-            transport=httpx.AsyncHTTPTransport(pool=self.custom_pool),
-            timeout=30.0,
-            verify=self.ssl_context,
+            transport=httpx.AsyncHTTPTransport(
+                retries=3,
+                verify=self.ssl_context
+            ),
+            timeout=30.0
         )
 
     def initialize_mms_model(self):