Update main.py

main.py

@@ -18,6 +18,7 @@ started_rooms = {}
 complete = {}
 openai.api_key='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'
 
+
 def generate_room_code():
     return ''.join(random.choices(string.ascii_uppercase, k=6))
 
@@ -25,7 +26,8 @@ def generate_room_code():
 @app.route('/create-room', methods=['POST'])
 def create_room():
     room_code = generate_room_code()
-    rooms[room_code] = {'players': [], 'started': False, 'process': 0, 'generated': False, 'story_content': '', 'reactions': 0, 'chats': [], 'pollCount': 0}
+    rooms[room_code] = {'players': [], 'started': False, 'process': 0, 'generated': False, 'story_content': '',
+                        'reactions': 0, 'chats': [], 'pollCount': 0}
 
     player_name = request.json.get('name')
     if not player_name:
@@ -44,7 +46,7 @@ def join_room(room_code):
         if room_code not in rooms:
             return jsonify({'noRoom': True})
 
-        if len(rooms[room_code]['players']) >= 2:
+        if len(rooms[room_code]['players']) >= 5:
             return jsonify({'error': 'Room is full'})
 
         player_name = request.json.get('name')
@@ -63,10 +65,11 @@ def join_room(room_code):
 
         ready = False
 
-        if len(rooms[room_code]['players']) == 2:
+        if len(rooms[room_code]['players']) == 5:  # do it <= 5
             ready = True
 
-        return jsonify({'players': rooms[room_code]['players'], 'ready': ready, 'started': rooms[room_code]['started'], 'meta_data': rooms[room_code]})
+        return jsonify({'players': rooms[room_code]['players'], 'ready': ready, 'started': rooms[room_code]['started'],
+                        'meta_data': rooms[room_code]})
 
 
 @app.route('/start-room/<room_code>', methods=['POST'])
@@ -91,13 +94,19 @@ def handle_tokens(room_code, player_id):
         genres = data['genres']
         additional_changes = data['additionalChanges']
 
-        rooms[room_code]['players'] = [{**player, 'word': word, 'genres': genres, 'additional_changes': additional_changes} if player['id'] == player_id else player for player in rooms[room_code]['players']]
+        rooms[room_code]['players'] = [
+            {**player, 'word': word, 'genres': genres, 'additional_changes': additional_changes} if player[
+                                                                                                        'id'] == player_id else player
+            for player in rooms[room_code]['players']]
         rooms[room_code]['process'] += 1
 
-        if rooms[room_code]['process'] == 2:
+        if rooms[room_code]['process'] == 5:
             process(room_code)
             return jsonify({'success': True, 'process': True})
 
+        print(rooms[room_code]['players'])
+        print(rooms[room_code]['process'])
+
         return jsonify({'success': True, 'process': False})
 
     if request.method == 'GET':
@@ -107,7 +116,7 @@ def handle_tokens(room_code, player_id):
         processed = False
         generated = False
 
-        if rooms[room_code]['process'] == 2:
+        if rooms[room_code]['process'] == 5:
             processed = True
 
         if rooms[room_code]['generated']:
@@ -154,11 +163,27 @@ def process(room_code):
             room_data[room_code]['additional_changes'] = player['additional_changes']
 
     # Find the most repeated genre in the room
+    # genre_counts = {}
+    # for player in players:
+    #     for genre in player['genres']:
+    #         genre_counts[genre] = genre_counts.get(genre, 0) + 1
+    # most_repeated_genre = max(genre_counts, key=genre_counts.get)
+    # room_data[room_code]['most_repeated_genre'] = most_repeated_genre
+
     genre_counts = {}
     for player in players:
         for genre in player['genres']:
-            genre_counts[genre] = genre_counts.get(genre, 0) + 1
-    most_repeated_genre = max(genre_counts, key=genre_counts.get)
+            if genre in genre_counts:
+                genre_counts[genre] += 1
+            else:
+                genre_counts[genre] = 1
+    most_repeated_genre = ''
+    maxct = 0
+    for key, value in genre_counts.items():
+        if maxct < value:
+            most_repeated_genre = key
+            maxct = value
+
     room_data[room_code]['most_repeated_genre'] = most_repeated_genre
 
     print(room_data)
@@ -169,15 +194,36 @@ def generate_story(room_code):
     if room_code not in rooms:
         return jsonify({'error': 'Room not found'})
 
-    if rooms[room_code]['process'] != 2:
-        return jsonify({'error': 'Room not ready'})
+    data = request.get_json()
+    word_limit = data.get('wordLimit')
+    word_data = room_data[room_code]['word_data']
+    additional_context = room_data[room_code]['additional_changes']
+    genre = room_data[room_code]['most_repeated_genre']
+    words = ''
+    ct = 0
+    for key, value in word_data.items():
+        ct += 1
+        if ct < len(word_data):
+            words += (value + ', ')
+        else:
+            words += value
+
+    if len(additional_context.split()) >= 250:
+        additional_context = ''
+
+    prompt = 'Keyword: ' + str(words) + '.' + 'Genres: ' + str(genre) + '.' + 'Additional Context: ' + str(
+        additional_context) + '.' + 'Word Limit: ' + str(word_limit) + '.'
+
+    print(prompt)
 
-    {'JFLMHJ': {'word_data': {'word 1': 'Word 1', 'word 2': 'word 2'},'additional_changes': 'Paragraphs should be 4. Paragraph 3', 'most_repeated_genre':'Fantasy'}}
+    profanity.load_censor_words()
 
-    prompt = 'Keyword: '+ 'word' + '.' + 'Genres: ' + 'genres' + '.' + 'Additional Context: ' + 'additional' + '.'
+    if profanity.contains_profanity(prompt):
+        return jsonify({'success': False, 'message': 'Content is improper.'})
 
     conversation = [
-        {"role": "system", "content": "You are a creative storyteller. User will give atmost 5 Keywords along with genres and additional context for the story."},
+        {"role": "system",
+         "content": "You are a creative storyteller. User will give atmost 5 Keywords along with genres and additional context for the story. Also user will mention the word limit for the story."},
         {"role": "user", "content": prompt}
     ]
 
@@ -191,13 +237,13 @@ def generate_story(room_code):
     story = response['choices'][0]['message']['content']
 
     rooms[room_code]['generated'] = True
-    rooms[room_code]['story_content'] = 'Meow'
+    rooms[room_code]['story_content'] = story
 
     print(rooms[room_code])
 
     complete[room_code] = rooms[room_code]
 
-    return jsonify({'success': True, 'storyContent': 'Meow Meow'})
+    return jsonify({'success': True, 'storyContent': story})
 
 
 @app.route('/reaction-true/<room_code>', methods=['GET'])