:lipstick: formatting for GUI

app.py

@@ -66,7 +66,7 @@ def ask_gpt(message: str):
     Returns:
         [str]: [model response as a string]
     """
-    st = time.time()
+    st = time.perf_counter()
     prompt = clean(message)  # clean user input
     prompt = prompt.strip()  # get rid of any extra whitespace
     if len(prompt) > 200:
@@ -81,12 +81,12 @@ def ask_gpt(message: str):
         top_p=0.65,  # optimize this with hyperparam search
     )
     bot_resp = gramformer_correct(corrector, qphrase=resp["out_text"])
-    rt = round(time.time() - st, 2)
+    rt = round(time.perf_counter() - st, 2)
     print(f"took {rt} sec to respond")
     return bot_resp
 
 
-def chat(message):
+def chat(trivia_query):
     """
     chat - helper function that makes the whole gradio thing work.
 
@@ -98,8 +98,8 @@ def chat(message):
         [str]: [returns an html string to display]
     """
     history = []
-    response = ask_gpt(message)
-    history.append(f"<b>{message}</b> <br> <br> <b>{response}</b>")
+    response = ask_gpt(trivia_query)
+    history.append(f"<b>Trivia Question: {trivia_query}</b> <br> <br> <b>Ballpark Answer: {response}</b>")
     gr.set_state(history)  # save the history
     html = ""
     for item in history:
@@ -150,7 +150,7 @@ if __name__ == "__main__":
         chat,
         inputs=["text"],
         outputs="html",
-        title=f"[WORK-IN-PROGRESS] Ballpark Trivia: {default_model} Model",
+        title=f"Ballpark Trivia: {default_model} Model",
         description=f"Are you frequently asked google-able Trivia questions and annoyed by it? Well, this is the app for you! Ballpark Trivia Bot answers any trivia question with something that sounds plausible but is probably not 100% correct. \n\n One might say.. the answers are in the right ballpark.",
         article="Further details can be found in the [model card](https: // huggingface.co/pszemraj/Ballpark-Trivia-L).  If you are interested in a more deceptively incorrect model, there is also an XL version on my page.\n\n"
         "**Important Notes & About:**\n\n"
@@ -171,5 +171,5 @@ if __name__ == "__main__":
     # launch the gradio interface and start the server
     iface.launch(
         share=True,
-        enable_queue=True,
+        enable_queue=True, # also allows for dealing with multiple users simultaneously (per newer gradio version)
     )