feat: get the username of the user

app.py

@@ -19,6 +19,7 @@ PUSH_FREQUENCY = 60 # every minute
 
 HISTORY = ""
 PROMPT = ""
+USERNAME = ""
 
 DATA_FILENAME = "data.jsonl"
 DATA_FILE = os.path.join("data", DATA_FILENAME)
@@ -91,7 +92,9 @@ def format_prompt(message, history, system_prompt):
   return prompt
 
 def generate(
-    prompt, history, system_prompt, temperature=0.9, max_new_tokens=100, top_p=0.95, repetition_penalty=1.0, seed=42,
+    prompt, history, system_prompt, 
+    temperature=0.9, max_new_tokens=100, top_p=0.95, 
+    repetition_penalty=1.0, seed=42,
 ):
     global HISTORY
     HISTORY = history
@@ -227,7 +230,7 @@ def vote(data: gr.LikeData):
                     {
                         "time_stamp": time.time(),
                         "model_version":MODEL_VERSION,
-                        "username":"",
+                        "username":USERNAME,
                         "prompt": PROMPT,
                         "history": HISTORY,
                         "system prompt": additional_inputs[0].value,
@@ -242,7 +245,16 @@ def vote(data: gr.LikeData):
         ]
         jsonlfile.write("\n".join(json_data) + "\n")
 
+def create_greeting(request: gr.Request):
+    global USERNAME
+    USERNAME = request.username
+    return gr.Markdown.update(value=f"Thanks for logging in, {request.username}")
+
 with gr.Blocks(css=CSS) as demo:
+
+    user_ip = gr.Markdown(value="Not logged in")
+    demo.load(create_greeting, inputs=None, outputs=user_ip)
+    
     with gr.Row():
         with gr.Column():
             gr.Markdown(