Update main.py

main.py

@@ -192,7 +192,7 @@ def process(room_code):
 @app.route('/generate-story/<room_code>', methods=['POST'])
 def generate_story(room_code):
     if room_code not in rooms:
-        return jsonify({'error': 'Room not found'})
+        return {'error': 'Room not found'}
 
     data = request.get_json()
     word_limit = data.get('wordLimit')
@@ -219,7 +219,7 @@ def generate_story(room_code):
     profanity.load_censor_words()
 
     if profanity.contains_profanity(prompt):
-        return jsonify({'success': False, 'message': 'Content is improper.'})
+        return {'success': False, 'message': 'Content is improper.'}
 
     conversation = [
         {"role": "system",
@@ -243,7 +243,7 @@ def generate_story(room_code):
 
     complete[room_code] = rooms[room_code]
 
-    return jsonify({'success': True, 'storyContent': story})
+    return {'success': True, 'storyContent': story}
 
 
 @app.route('/reaction-true/<room_code>', methods=['GET'])
@@ -294,7 +294,7 @@ def completed(room_code):
     if room_code not in rooms:
         return jsonify({'success': False, 'error': 'Room not found'})
 
-    return jsonify(complete[room_code])
+    return complete[room_code]
 
 
 @app.route('/list-rooms', methods=['GET'])
@@ -303,7 +303,7 @@ def list_rooms():
     for room_code, player in rooms.items():
         room_info[room_code] = len(player['players'])
 
-    return jsonify({'roomInfo': room_info})
+    return {'roomInfo': room_info}
 
 
 if __name__ == '__main__':