Update app.py

app.py

@@ -1,42 +1,35 @@
 import flask
 import flask_cors
+import os
+
+message = {"123": {}}
 
 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():
-    data = flask.request.json
-    name_in = data.get('nameInput')
-    message_in = data.get('messageInput')
-    room_id = data.get('roomId')
-
-    if not all((name_in, message_in, room_id)):
-        return flask.jsonify({'error': 'Invalid data'})
-
-    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})
+    
+    nameIn = flask.request.get_json()['nameInput']
+    messageIn = flask.request.get_json()['messageInput']
+    roomidIn = flask.request.get_json()['roomIdInput']
+    
+    n = len(message[roomidIn])+1
+    
+    message[roomidIn].update({"message"+str(n): {"name": nameIn, "message": messageIn}})
+    
+    return flask.jsonify("success": True)
 
 @app.route('/messages', methods=['POST'])
 def messages():
-    data = flask.request.json
-    room_id = data.get('roomId')
+    roomidInput = flask.request.get_json()['roomIdInput']
+    return flask.jsonify(message[roomidInput])
+
 
-    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)