Make the code more like PEP8 for readability (#862)

modules/html_generator.py

@@ -24,6 +24,7 @@ with open(Path(__file__).resolve().parent / '../css/html_cai_style.css', 'r') as
 with open(Path(__file__).resolve().parent / '../css/html_instruct_style.css', 'r') as f:
     instruct_css = f.read()
 
+
 def fix_newlines(string):
     string = string.replace('\n', '\n\n')
     string = re.sub(r"\n{3,}", "\n\n", string)
@@ -31,6 +32,8 @@ def fix_newlines(string):
     return string
 
 # This could probably be generalized and improved
+
+
 def convert_to_markdown(string):
     string = string.replace('\\begin{code}', '```')
     string = string.replace('\\end{code}', '```')
@@ -38,13 +41,15 @@ def convert_to_markdown(string):
     string = string.replace('\\end{blockquote}', '')
     string = re.sub(r"(.)```", r"\1\n```", string)
     string = fix_newlines(string)
-    return markdown.markdown(string, extensions=['fenced_code']) 
+    return markdown.markdown(string, extensions=['fenced_code'])
+
 
 def generate_basic_html(string):
     string = convert_to_markdown(string)
     string = f'<style>{readable_css}</style><div class="container">{string}</div>'
     return string
 
+
 def process_post(post, c):
     t = post.split('\n')
     number = t[0].split(' ')[1]
@@ -59,6 +64,7 @@ def process_post(post, c):
     src = f'<span class="name">Anonymous </span> <span class="number">No.{number}</span>\n{src}'
     return src
 
+
 def generate_4chan_html(f):
     posts = []
     post = ''
@@ -84,7 +90,7 @@ def generate_4chan_html(f):
             posts[i] = f'<div class="op">{posts[i]}</div>\n'
         else:
             posts[i] = f'<div class="reply">{posts[i]}</div>\n'
-    
+
     output = ''
     output += f'<style>{_4chan_css}</style><div id="parent"><div id="container">'
     for post in posts:
@@ -98,13 +104,15 @@ def generate_4chan_html(f):
 
     return output
 
+
 def make_thumbnail(image):
-    image = image.resize((350, round(image.size[1]/image.size[0]*350)), Image.Resampling.LANCZOS)
+    image = image.resize((350, round(image.size[1] / image.size[0] * 350)), Image.Resampling.LANCZOS)
     if image.size[1] > 470:
         image = ImageOps.fit(image, (350, 470), Image.ANTIALIAS)
 
     return image
 
+
 def get_image_cache(path):
     cache_folder = Path("cache")
     if not cache_folder.exists():
@@ -119,9 +127,10 @@ def get_image_cache(path):
 
     return image_cache[path][1]
 
+
 def generate_instruct_html(history):
     output = f'<style>{instruct_css}</style><div class="chat" id="chat">'
-    for i,_row in enumerate(history[::-1]):
+    for i, _row in enumerate(history[::-1]):
         row = [convert_to_markdown(entry) for entry in _row]
 
         output += f"""
@@ -134,7 +143,7 @@ def generate_instruct_html(history):
               </div>
             """
 
-        if len(row[0]) == 0: # don't display empty user messages
+        if len(row[0]) == 0:  # don't display empty user messages
             continue
 
         output += f"""
@@ -151,6 +160,7 @@ def generate_instruct_html(history):
 
     return output
 
+
 def generate_cai_chat_html(history, name1, name2, reset_cache=False):
     output = f'<style>{cai_css}</style><div class="chat" id="chat">'
 
@@ -159,7 +169,7 @@ def generate_cai_chat_html(history, name1, name2, reset_cache=False):
     img_bot = f'<img src="file/cache/pfp_character.png{suffix}">' if Path("cache/pfp_character.png").exists() else ''
     img_me = f'<img src="file/cache/pfp_me.png{suffix}">' if Path("cache/pfp_me.png").exists() else ''
 
-    for i,_row in enumerate(history[::-1]):
+    for i, _row in enumerate(history[::-1]):
         row = [convert_to_markdown(entry) for entry in _row]
 
         output += f"""
@@ -178,7 +188,7 @@ def generate_cai_chat_html(history, name1, name2, reset_cache=False):
               </div>
             """
 
-        if len(row[0]) == 0: # don't display empty user messages
+        if len(row[0]) == 0:  # don't display empty user messages
             continue
 
         output += f"""
@@ -200,9 +210,11 @@ def generate_cai_chat_html(history, name1, name2, reset_cache=False):
     output += "</div>"
     return output
 
+
 def generate_chat_html(history, name1, name2):
     return generate_cai_chat_html(history, name1, name2)
 
+
 def chat_html_wrapper(history, name1, name2, mode, reset_cache=False):
     if mode == "cai-chat":
         return generate_cai_chat_html(history, name1, name2, reset_cache)