Update app.py

app.py

@@ -3,43 +3,32 @@ from datetime import datetime
 from gradio.components import *
 from gradio.inputs import *
 
-# Define chatrooms and versions
-CHATROOMS = {}
-VERSIONS = {
-    "app.py, version 1.0": "app.py",
-    "app_2.py, version 1.1": "app_2.py"
-}
+chatrooms = {}
 
-def chatroom(username, message, send_message, chatroom_name, image=None):
-    if chatroom_name not in CHATROOMS:
-        CHATROOMS[chatroom_name] = []
+def chatroom(username, message, image, send_message, chatroom_name):
+    if chatroom_name not in chatrooms:
+        chatrooms[chatroom_name] = []
     if send_message:
         now = datetime.now()
         timestamp = now.strftime("%Y-%m-%d %H:%M:%S")
-        CHATROOMS[chatroom_name].append((username, message, timestamp, image))
-    message_history = "<br>".join([f"{msg[2]} {msg[0]}: {msg[1]}" + (f"<br><img src='{msg[3]}' width='200'>" if msg[3] else "") for msg in CHATROOMS[chatroom_name]])
+        chatrooms[chatroom_name].append((username, message, image, timestamp))
+    message_history = ""
+    for msg in chatrooms[chatroom_name]:
+        if msg[1]:
+            message_history += f"{msg[3]} {msg[0]}: {msg[1]}<br>"
+        if msg[2]:
+            message_history += f"{msg[3]} {msg[0]}: <img src='data:image/png;base64,{msg[2]}' /><br>"
     return message_history, ""
 
-# Define inputs and outputs
-inputs = [
-    Textbox("Username"),
-    Textbox("Message"),
-    Checkbox("Send"),
-    Textbox(label="Chatroom Name"),
-    Image(type="numpy", label="Image (optional)", optional=True),
-    choose("Version", choices=list(VERSIONS.keys()))
-]
-outputs = ["html", "text"]
-
-# Define interface
-iface = gr.Interface(fn=chatroom,
-                     inputs=inputs,
-                     outputs=outputs,
+iface = gr.Interface(fn=chatroom, 
+                     inputs=[Textbox("Username"), 
+                             Textbox("Message"), 
+                             Image(shape=(224, 224), image_mode="RGB", source="upload", label="Image"), 
+                             Checkbox("Send"), 
+                             Textbox(label="Chatroom Name")],
+                     outputs=["html", "text"],
                      layout="vertical",
                      title="Public Chatroom",
-                     description="Type your username and message, and click Send to send to the chatroom.")
+                     description="Type your username, message, and/or upload an image, and click Send to send to the chatroom.")
 
-# Define file paths
-version = iface.inputs[-1].value
-iface.share(file_name=VERSIONS[version])
 iface.launch()