From f616dcc81a3e63536bffc6d250118baa3d230e5f Mon Sep 17 00:00:00 2001
From: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
Date: Sun, 12 Feb 2023 18:32:20 +0000
Subject: [PATCH] Added handling of missing aiohttp module

---
 main.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index ba2040599..0c1b37c77 100644
--- a/main.py
+++ b/main.py
@@ -7,6 +7,16 @@ import heapq
 import traceback
 import asyncio
 
+try:
+    import aiohttp
+    from aiohttp import web
+except ImportError:
+    print("Module 'aiohttp' not installed. Please install it via:")
+    print("pip install aiohttp")
+    print("or")
+    print("pip install -r requirements.txt")
+    sys.exit()
+
 if __name__ == "__main__":
     if '--help' in sys.argv:
         print("Valid Command line Arguments:")
@@ -346,9 +356,6 @@ class PromptQueue:
                     return True
         return False
 
-import aiohttp
-from aiohttp import web
-
 def get_queue_info(prompt_queue):
     prompt_info = {}
     exec_info = {}