Update app.py

app.py

@@ -1,32 +1,42 @@
 import flask
 import flask_cors
-import os
-
-messageArray = {}
 
 app = flask.Flask(__name__)
 flask_cors.CORS(app)
 
+rooms = {}
+
 @app.route('/')
 def index():
     return flask.render_template('index.html')
 
 @app.route('/postMessage', methods=['POST'])
 def post():
-    nameIn = flask.request.get_json()['nameInput']
-    messageIn = flask.request.get_json()['messageInput']
+    data = flask.request.json
+    name_in = data.get('nameInput')
+    message_in = data.get('messageInput')
+    room_id = data.get('roomId')
 
-    n = len(messageArray)+1
-    
-    messageArray.update({"message"+str(n): {"name": nameIn, "message": messageIn}})
-    
-    return flask.jsonify("Message Sent")
+    if not all((name_in, message_in, room_id)):
+        return flask.jsonify({'error': 'Invalid data'})
 
-@app.route('/messages', methods=['GET'])
-def messages():
-    return flask.jsonify(messageArray)
+    if room_id not in rooms:
+        rooms[room_id] = []
 
+    n = len(rooms[room_id]) + 1
+    rooms[room_id].append({"message" + str(n): {"name": name_in, "message": message_in}})
+
+    return flask.jsonify({'success': True})
+
+@app.route('/messages', methods=['POST'])
+def messages():
+    data = flask.request.json
+    room_id = data.get('roomId')
 
+    if room_id in rooms:
+        return flask.jsonify(rooms[room_id])
+    else:
+        return flask.jsonify({"error": "Room not found"})
 
 if __name__ == "__main__":
-    app.run(debug=True,host="0.0.0.0",port=5000)
+    app.run(debug=True, host="0.0.0.0", port=5000)