Update app.py

app.py

@@ -4,14 +4,19 @@ from urllib.request import urlopen
 import json
 import time
 
+
+# populate all models available from GPT4All
 url = "https://raw.githubusercontent.com/nomic-ai/gpt4all/main/gpt4all-chat/metadata/models3.json"
 response = urlopen(url)
 data_json = json.loads(response.read())
 
+
 def model_choices():
     model_list = [data_json[i]['filename'] for i in range(len(data_json))]
     return model_list
 
+    
+# get each models' description
 model_description = {model['filename']: model['description'] for model in data_json}
 
 
@@ -20,7 +25,39 @@ def llm_intro(selected_model):
     formatted_description = html_string.replace("<ul>", "").replace("</ul>", "").replace("</li>", "").replace("<br>", "\n").replace("</br>", "").replace("<li>", "\n➤ ")
     return formatted_description
 
-model_cache = {}  # Global cache
+
+def remove_endtags(html_string, tags):
+    """Remove rear HTML tags from the input string."""
+    for tag in tags:
+        html_string = re.sub(fr"</{tag}>", "", html_string)
+    return html_string
+
+
+def replace_starttags(html_string, replacements):
+    """Replace starting HTML tags with the corresponding values."""
+    for tag, replacement in replacements.items():
+        html_string = html_string.replace(tag, replacement)
+    return html_string
+
+
+def format_html_string(html_string):
+    """Format the HTML string to a readable text format."""
+    tags_to_remove = ["ul", "li", "br"]
+    html_string = remove_endtags(html_string, tags_to_remove)
+
+    tag_replacements = {
+        "<li>": "\n➤ ",
+        "<br>": "\n",
+        "<strong>": "**",
+        "</strong>": "**"
+    }
+    formatted_string = replace_starttags(html_string, tag_replacements)
+
+    return formatted_string
+
+    
+# cache models for faster reloads
+model_cache = {}  
 
 
 def load_model(model_name):
@@ -36,7 +73,7 @@ def load_model(model_name):
 
 #     clear = gr.ClearButton([input_text, chatbot])
 
-# Define the chatbot function
+# Construct chatbot
 def generate_response(model_name, message, chat_history):
     model = load_model(model_name)
     chat_history = []
@@ -49,7 +86,7 @@ def generate_response(model_name, message, chat_history):
     chat_history.append((input_text, response))
     return chat_history, response
 
-# Create the Gradio interface
+# Create Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# GPT4All Chatbot")
     with gr.Row():