Use demo.load

app.py

@@ -11,31 +11,38 @@ DATA_FILE = os.path.join("/data", DATA_FILENAME)
 MAX_LINES = 20
 
 def generate_html(me: str) -> str:
-    with open(DATA_FILE) as csvfile:
-        reader = csv.reader(csvfile)
-        rows = deque(reader, maxlen=MAX_LINES)
-        if len(rows) == 0:
-            return "no messages yet"
-        else:
-            html = "<div class='chatbot'>"
-            for row, _ in itertools.zip_longest(rows, range(MAX_LINES), fillvalue=("", "")):
-                username = row[0]
-                if username == me:
-                    msg_type = "own"
-                elif username == "[chatbot]":
-                    msg_type = "admin"
-                elif username == "":
-                    msg_type = "empty"
-                else:
-                    msg_type = "other"
-                html += "<div class='message'>"
-                html += f"<span class='nick'>{row[0]}</span>"
-                html += f"<span class='{msg_type}'>{row[1]}</span>"
+    try:
+        with open(DATA_FILE) as csvfile:
+            reader = csv.reader(csvfile)
+            rows = deque(reader, maxlen=MAX_LINES)
+            if len(rows) == 0:
+                return "no messages yet"
+            else:
+                html = "<div class='chatbot'>"
+                for row, _ in itertools.zip_longest(rows, range(MAX_LINES), fillvalue=("", "")):
+                    username = row[0]
+                    if username == me:
+                        msg_type = "own"
+                    elif username == "[chatbot]":
+                        msg_type = "admin"
+                    elif username == "":
+                        msg_type = "empty"
+                    else:
+                        msg_type = "other"
+                    html += "<div class='message'>"
+                    html += f"<span class='nick'>{row[0]}</span>"
+                    html += f"<span class='{msg_type}'>{row[1]}</span>"
+                    html += "</div>"
                 html += "</div>"
-            html += "</div>"
-            return html
+                return html
+    except:
+        return "no messages yet"
 
 
+def refresh(state):
+    print(f"refresh, username: {state['username']}")
+    return generate_html(state["username"])
+
 def store_message(writer: str, message: str, me: str):
     if writer and message:
         with open(DATA_FILE, "a") as csvfile:
@@ -80,15 +87,14 @@ with gr.Blocks(css=css) as demo:
     gr.Markdown(
         """
         <h1><center>Persistent Storage Chat</center></h1>
-        <p>This is a simple demo that implements a chatroom using persistent storage.
-        All the messages are saved in a file, but only the last 20 ones are displayed.</p>
+        <p>This is a simple demo that implements a chatroom using persistent storage.</p>
         """
     )
 
     with gr.Group():
         with (login_box := gr.Row().style(equal_height=True)):
             username = gr.Textbox(
-                label="What's your name? (leave empty for anonymous)", show_label=True, max_lines=1
+                label="What's your name?", show_label=True, max_lines=1
             )
             login_btn = gr.Button("Enter the chat")
 
@@ -97,7 +103,7 @@ with gr.Blocks(css=css) as demo:
             output = gr.HTML(label="Chat")
         with gr.Row():
             message = gr.Textbox(
-                label="Your message (leave empty to refresh)", show_label=True, max_lines=1
+                label="Your message", show_label=True, max_lines=1
             )
 
     username.submit(
@@ -116,4 +122,5 @@ with gr.Blocks(css=css) as demo:
         inputs=[message, state],
         outputs=[output, message],
     )
-demo.launch()
+    demo.load(refresh, queue=True, inputs=[state], outputs=output, every=5)
+demo.queue().launch(server_name="0.0.0.0")