PEP8 code styling, no functional change

Author: dasmy

gpt_code_ui/kernel_program/config.py

@@ -15,4 +15,4 @@ def get_logger():
     logger = logging.getLogger(__name__)
     if "DEBUG" in os.environ:
         logger.setLevel(logging.DEBUG)
-    return logger
+    return logger

gpt_code_ui/kernel_program/kernel_manager.py

@@ -56,7 +56,7 @@ def cleanup_spawned_processes():
                         os.kill(pid, signal.CTRL_BREAK_EVENT)
                     else:
                         os.kill(pid, signal.SIGKILL)
-                    
+
                     # After successful kill, cleanup pid file
                     os.remove(fp)
 
@@ -221,4 +221,4 @@ def start_kernel():
 
 if __name__ == "__main__":
     kc = start_kernel()
-    start_snakemq(kc)
+    start_snakemq(kc)

gpt_code_ui/kernel_program/launch_kernel.py

@@ -1,4 +1,4 @@
 if __name__ == "__main__":
     from ipykernel import kernelapp as app
 
-    app.launch_new_instance()
+    app.launch_new_instance()

gpt_code_ui/kernel_program/main.py

@@ -7,7 +7,6 @@
 import time
 
 import asyncio
-import json
 import threading
 
 from queue import Queue
@@ -47,6 +46,7 @@
 app = Flask(__name__)
 CORS(app)
 
+
 def start_kernel_manager():
     global kernel_manager_process
 
@@ -62,9 +62,11 @@ def start_kernel_manager():
     with open(os.path.join(config.KERNEL_PID_DIR, "%d.pid" % kernel_manager_process.pid), "w") as p:
         p.write("kernel_manager")
 
+
 def cleanup_kernel_program():
     kernel_manager.cleanup_spawned_processes()
 
+
 async def start_snakemq():
     global messaging
 
@@ -77,7 +79,7 @@ def on_recv(conn, ident, message):
             if message["value"] == "ready":
                 logger.debug("Kernel is ready.")
                 result_queue.put({
-                    "value":"Kernel is ready.",
+                    "value": "Kernel is ready.",
                     "type": "message"
                 })
 
@@ -97,8 +99,9 @@ def send_queued_messages():
         while True:
             if send_queue.qsize() > 0:
                 message = send_queue.get()
-                utils.send_json(messaging, 
-                    {"type": "execute", "value": message["command"]}, 
+                utils.send_json(
+                    messaging,
+                    {"type": "execute", "value": message["command"]},
                     config.IDENT_KERNEL_MANAGER
                 )
             time.sleep(0.1)
@@ -117,7 +120,7 @@ async def async_link_loop():
 
 @app.route("/api", methods=["POST", "GET"])
 def handle_request():
-   
+
     if request.method == "GET":
         # Handle GET requests by sending everything that's in the receive_queue
         results = [result_queue.get() for _ in range(result_queue.qsize())]
@@ -128,7 +131,8 @@ def handle_request():
         send_queue.put(data)
 
         return jsonify({"result": "success"})
-    
+
+
 @app.route("/restart", methods=["POST"])
 def handle_restart():
 
@@ -152,9 +156,6 @@ async def main():
 def run_flask_app():
     app.run(host="0.0.0.0", port=APP_PORT)
 
+
 if __name__ == "__main__":
     asyncio.run(main())
-
-
-
-    

gpt_code_ui/kernel_program/utils.py

@@ -7,6 +7,7 @@
 
 import gpt_code_ui.kernel_program.config as config
 
+
 def escape_ansi(line):
     ansi_escape = re.compile(r"(?:\x1B[@-_]|[\x80-\x9F])[0-?]*[ -/]*[@-~]")
     return ansi_escape.sub("", line)
@@ -16,6 +17,7 @@ def send_json(messaging, message, identity):
     message = snakemq.message.Message(json.dumps(message).encode("utf-8"), ttl=600)
     messaging.send_message(identity, message)
 
+
 def init_snakemq(ident, init_type="listen"):
     link = snakemq.link.Link()
     packeter = snakemq.packeter.Packeter(link)
@@ -26,4 +28,4 @@ def init_snakemq(ident, init_type="listen"):
         link.add_connector(("localhost", config.SNAKEMQ_PORT))
     else:
         raise Exception("Unsupported init type.")
-    return messaging, link
+    return messaging, link

gpt_code_ui/main.py

@@ -16,20 +16,23 @@
 
 APP_URL = "http://localhost:%s" % APP_PORT
 
+
 def run_webapp():
     try:
         app.run(host="0.0.0.0", port=APP_PORT, use_reloader=False)
-    except Exception as e:
-        logging.exception("Error running the webapp:")
+    except Exception:
+        logging.exception("Error running the webapp")
         sys.exit(1)
 
+
 def run_kernel_program():
     try:
         asyncio.run(kernel_program_main())
-    except Exception as e:
-        logging.exception("Error running the kernel_program:")
+    except Exception:
+        logging.exception("Error running the kernel_program")
         sys.exit(1)
 
+
 def setup_logging():
     log_format = "%(asctime)s [%(levelname)s]: %(message)s"
     logging.basicConfig(level=logging.INFO, format=log_format)
@@ -38,32 +41,33 @@ def setup_logging():
     file_handler.setFormatter(logging.Formatter(log_format))
     logging.getLogger().addHandler(file_handler)
 
+
 def print_color(text, color="gray"):
     # Default to gray
-    code="242"
+    code = "242"
 
     if color == "green":
-        code="35"
-    
+        code = "35"
+
     gray_code = "\033[38;5;%sm" % code
     reset_code = "\033[0m"
     print(f"{gray_code}{text}{reset_code}")
 
 
 def print_banner():
-        
-        print("""
+    print("""
 █▀▀ █▀█ ▀█▀ ▄▄ █▀▀ █▀█ █▀▄ █▀▀
 █▄█ █▀▀ ░█░ ░░ █▄▄ █▄█ █▄▀ ██▄
-        """)
+    """)
+
+    print("> Open GPT-Code UI in your browser %s" % APP_URL)
+    print("")
+    print("You can inspect detailed logs in app.log.")
+    print("")
+    print("Find your OpenAI API key at https://platform.openai.com/account/api-keys")
+    print("")
+    print_color("Contribute to GPT-Code UI at https://github.com/ricklamers/gpt-code-ui")
 
-        print("> Open GPT-Code UI in your browser %s" % APP_URL)
-        print("")
-        print("You can inspect detailed logs in app.log.")
-        print("")
-        print("Find your OpenAI API key at https://platform.openai.com/account/api-keys")
-        print("")
-        print_color("Contribute to GPT-Code UI at https://github.com/ricklamers/gpt-code-ui")   
 
 def main():
     setup_logging()
@@ -80,20 +84,19 @@ def main():
             try:
                 app.test_client().get("/")
                 break
-            except:
+            except Exception:
                 time.sleep(0.1)
-        
-        print_banner()    
-        
+
+        print_banner()
+
         webbrowser.open(APP_URL)
 
         webapp_process.join()
         kernel_program_process.join()
 
-        
     except KeyboardInterrupt:
         print("Terminating processes...")
-        
+
         cleanup_kernel_program()
         kernel_program_process.terminate()
 
@@ -103,6 +106,7 @@ def main():
         kernel_program_process.join()
 
         print("Processes terminated.")
-        
+
+
 if __name__ == '__main__':
     main()

gpt_code_ui/webapp/main.py

@@ -94,34 +94,34 @@ def inspect_file(filename: str) -> str:
 
 async def get_code(user_prompt, user_openai_key=None, model="gpt-3.5-turbo"):
 
-    prompt = f"""First, here is a history of what I asked you to do earlier. 
-    The actual prompt follows after ENDOFHISTORY. 
-    History:
-    {message_buffer.get_string()}
-    ENDOFHISTORY.
-    Write Python code, in a triple backtick Markdown code block, that does the following:
-    {user_prompt}
-    
-    Notes: 
-        First, think step by step what you want to do and write it down in English.
-        Then generate valid Python code in a code block 
-        Make sure all code is valid - it be run in a Jupyter Python 3 kernel environment. 
-        Define every variable before you use it.
-        For data munging, you can use 
-            'numpy', # numpy==1.24.3
-            'dateparser' #dateparser==1.1.8
-            'pandas', # matplotlib==1.5.3
-            'geopandas' # geopandas==0.13.2
-        For pdf extraction, you can use
-            'PyPDF2', # PyPDF2==3.0.1
-            'pdfminer', # pdfminer==20191125
-            'pdfplumber', # pdfplumber==0.9.0
-        For data visualization, you can use
-            'matplotlib', # matplotlib==3.7.1
-        Be sure to generate charts with matplotlib. If you need geographical charts, use geopandas with the geopandas.datasets module.
-        If the user has just uploaded a file, focus on the file that was most recently uploaded (and optionally all previously uploaded files)
-    
-    Teacher mode: if the code modifies or produces a file, at the end of the code block insert a print statement that prints a link to it as HTML string: <a href='/download?file=INSERT_FILENAME_HERE'>Download file</a>. Replace INSERT_FILENAME_HERE with the actual filename."""
+    prompt = f"""First, here is a history of what I asked you to do earlier.
+The actual prompt follows after ENDOFHISTORY.
+History:
+{message_buffer.get_string()}
+ENDOFHISTORY.
+Write Python code, in a triple backtick Markdown code block, that does the following:
+{user_prompt}
+
+Notes:
+    First, think step by step what you want to do and write it down in English.
+    Then generate valid Python code in a code block
+    Make sure all code is valid - it be run in a Jupyter Python 3 kernel environment.
+    Define every variable before you use it.
+    For data munging, you can use
+        'numpy', # numpy==1.24.3
+        'dateparser' #dateparser==1.1.8
+        'pandas', # matplotlib==1.5.3
+        'geopandas' # geopandas==0.13.2
+    For pdf extraction, you can use
+        'PyPDF2', # PyPDF2==3.0.1
+        'pdfminer', # pdfminer==20191125
+        'pdfplumber', # pdfplumber==0.9.0
+    For data visualization, you can use
+        'matplotlib', # matplotlib==3.7.1
+    Be sure to generate charts with matplotlib. If you need geographical charts, use geopandas with the geopandas.datasets module.
+    If the user has just uploaded a file, focus on the file that was most recently uploaded (and optionally all previously uploaded files)
+
+Teacher mode: if the code modifies or produces a file, at the end of the code block insert a print statement that prints a link to it as HTML string: <a href='/download?file=INSERT_FILENAME_HERE'>Download file</a>. Replace INSERT_FILENAME_HERE with the actual filename."""
 
     if user_openai_key:
         openai.api_key = user_openai_key