Update main.py

main.py

@@ -1,58 +1,263 @@
-from flask import Flask, request
-import joblib
-import pandas as pd
+from email.message import EmailMessage
+from flask import Flask, request, jsonify, render_template
+import random
+import string
 from flask_cors import CORS
-
+import smtplib
+import bcrypt
+import openai
 
 app = Flask(__name__)
-app.static_folder = 'static'
-app.static_url_path = '/static'
+CORS(app)
 
-app.secret_key = "roadsense-abhi-2023"
 
-CORS(app)
+# Buffered Caches
+rooms = {}
+room_data = {}
+started_rooms = {}
+complete = {}
+openai.api_key='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'
+
+def generate_room_code():
+    return ''.join(random.choices(string.ascii_uppercase, k=6))
+
+
+@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}
+
+    player_name = request.json.get('name')
+    if not player_name:
+        return jsonify({'error': 'Player name is required'})
+
+    # Add the player to the room.
+    player_id = len(rooms[room_code]['players']) + 1
+    rooms[room_code]['players'].append({'id': player_id, 'name': player_name})
+
+    return jsonify({'room_code': room_code})
+
+
+@app.route('/join-room/<room_code>', methods=['POST', 'GET'])
+def join_room(room_code):
+    if request.method == 'POST':
+        if room_code not in rooms:
+            return jsonify({'noRoom': True})
+
+        if len(rooms[room_code]['players']) >= 2:
+            return jsonify({'error': 'Room is full'})
+
+        player_name = request.json.get('name')
+        if not player_name:
+            return jsonify({'error': 'Player name is required'})
+
+        # Add the player to the room.
+        player_id = len(rooms[room_code]['players']) + 1
+        rooms[room_code]['players'].append({'id': player_id, 'name': player_name})
+
+        return jsonify({'player_id': player_id, 'players': rooms[room_code]['players']})
+
+    if request.method == 'GET':
+        if room_code not in rooms:
+            return jsonify({'noRoom': True})
+
+        ready = False
+
+        if len(rooms[room_code]['players']) == 2:
+            ready = True
+
+        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'])
+def start_room(room_code):
+    if room_code not in rooms:
+        return jsonify({'error': 'Room not found'}), 404
+
+    rooms[room_code]['started'] = True
+    rooms[room_code]['process'] = 0
+
+    return jsonify({'success': True})
+
+
+@app.route('/handle-tokens/<room_code>/<int:player_id>', methods=['POST', 'GET'])
+def handle_tokens(room_code, player_id):
+    if request.method == 'POST':
+        if room_code not in rooms:
+            return jsonify({'error': 'Room not found'})
+
+        data = request.get_json()
+        word = data['word']
+        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]['process'] += 1
+
+        if rooms[room_code]['process'] == 2:
+            process(room_code)
+            return jsonify({'success': True, 'process': True})
+
+        return jsonify({'success': True, 'process': False})
+
+    if request.method == 'GET':
+        if room_code not in rooms:
+            return jsonify({'error': 'Room not found'})
+
+        processed = False
+        generated = False
+
+        if rooms[room_code]['process'] == 2:
+            processed = True
+
+        if rooms[room_code]['generated']:
+            generated = True
+
+        return jsonify({'process': processed, 'generated': generated})
+
+
+@app.route('/leave-room/<room_code>/<int:player_id>', methods=['POST'])
+def leave_room(room_code, player_id):
+    if room_code not in rooms:
+        return jsonify({'error': 'Room not found'})
+
+    rooms[room_code]['players'] = [player for player in rooms[room_code]['players'] if player['id'] != player_id]
+
+    if player_id == 1 or not rooms[room_code]['players']:
+        del rooms[room_code]
+
+    return jsonify({'message': 'Player has left the room'})
 
-    
-# Load the model
-model = joblib.load('accident_prediction_model_Final.m5')
 
-# Load the encoder
-encoder = joblib.load('encoder.pkl')
+def process(room_code):
+    if room_code not in rooms:
+        return jsonify({'error': 'Room not found'})
 
-@app.route('/', methods=['GET'])
-def main():
-    return {'message': 'Hello, World'}
+    players = rooms[room_code]['players']
 
+    # Initialize variables to store aggregated data
+    room_data[room_code] = {
+        'word_data': {},
+        'additional_changes': ''
+    }
 
-@app.route('/prediction', methods=['POST'])
-def prediction():
+    for player in players:
+        player_id = player['id']
+
+        # Store word data for each player
+        room_data[room_code]['word_data'][f'word {player_id}'] = player['word']
+
+        # Aggregate additional_changes
+        if room_data[room_code]['additional_changes']:
+            room_data[room_code]['additional_changes'] += ' ' + player['additional_changes']
+        else:
+            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
+
+    print(room_data)
+
+
+@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'})
+
+    if rooms[room_code]['process'] != 2:
+        return jsonify({'error': 'Room not ready'})
+
+    {'JFLMHJ': {'word_data': {'word 1': 'Word 1', 'word 2': 'word 2'},'additional_changes': 'Paragraphs should be 4. Paragraph 3', 'most_repeated_genre':'Fantasy'}}
+
+    prompt = 'Keyword: '+ 'word' + '.' + 'Genres: ' + 'genres' + '.' + 'Additional Context: ' + 'additional' + '.'
+
+    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": "user", "content": prompt}
+    ]
+
+    response = openai.ChatCompletion.create(
+        model='gpt-3.5-turbo',
+        messages=conversation,
+        temperature=0.7,
+        n=1
+    )
+
+    story = response['choices'][0]['message']['content']
+
+    rooms[room_code]['generated'] = True
+    rooms[room_code]['story_content'] = 'Meow'
+
+    print(rooms[room_code])
+
+    complete[room_code] = rooms[room_code]
+
+    return jsonify({'success': True, 'storyContent': 'Meow Meow'})
+
+
+@app.route('/reaction-true/<room_code>', methods=['GET'])
+def reaction_true(room_code):
+    rooms[room_code]['reactions'] += 1
+
+    return jsonify({'success': True, 'likeCount': rooms[room_code]['reactions']})
+
+
+@app.route('/reaction-false/<room_code>', methods=['GET'])
+def reaction_false(room_code):
+    rooms[room_code]['reactions'] -= 1
+
+    return jsonify({'success': True, 'likeCount': rooms[room_code]['reactions']})
+
+
+@app.route('/room-chat/<room_code>', methods=['GET'])
+def room_chat(room_code):
+    if room_code not in rooms:
+        return jsonify({'success': False, 'error': 'Room not found'})
+    print(rooms[room_code]['chats'])
+    return jsonify({'success': True, 'chats': rooms[room_code]['chats'], 'pollCount': rooms[room_code]['pollCount']})
+
+
+@app.route('/chat-input/<room_code>/<int:player_id>', methods=['POST'])
+def chat_input(room_code, player_id):
     data = request.get_json()
-    
-    num_input = {'Latitude': data['Latitude'], 'Longitude': data['Longitude'], 'person_count': data['personCount']}
-    cat_input = {'weather_conditions': data['selectedWeatherCondition'], 'impact_type': data['selectedImpactType'],
-                 'traffic_voilations': data['selectedTrafficViolationType'],
-                 'road_features': data['selectedRoadFeaturesType'],
-                 'junction_types': data['selectedRoadJunctionType'],
-                 'traffic_controls': data['selectedTrafficControl'], 'time_day': data['selectedTimeOfDay'],
-                 'age_group': data['selectedAge'], 'safety_features': data['selectedSafetyFeature'],
-                 'injury': data['selectedInjuryType']}
-
-    input_df = pd.DataFrame([cat_input])
-
-    encoded_input = encoder['encoder'].transform(input_df)
-    encoded_input_df = pd.DataFrame(encoded_input, columns=encoder['encoded_columns'])
-
-    num_df = pd.DataFrame([num_input])
-    input_with_coords = pd.concat([num_df, encoded_input_df], axis=1)
-
-    # Make a prediction using the trained model
-    prediction = model.predict(input_with_coords)
-
-    temp = False
-    if prediction[0] == 1:
-        temp = True
-    
-    return {'prediction': temp}
+    if room_code not in rooms:
+        return jsonify({'success': False, 'error': 'Room not found'})
+
+    rooms[room_code]['chats'].append({'id': player_id, 'text': data.get('text')})
+
+    return jsonify({'success': True})
+
+
+@app.route('/poll-data/<room_code>', methods=['POST'])
+def poll_count(room_code):
+    if room_code not in rooms:
+        return jsonify({'success': False, 'error': 'Room not found'})
+
+    rooms[room_code]['pollCount'] += 1
+
+    return jsonify({'success': True, 'pollCount': rooms[room_code]['pollCount']})
+
+
+@app.route('/complete/<room_code>', methods=['GET'])
+def completed(room_code):
+    if room_code not in rooms:
+        return jsonify({'success': False, 'error': 'Room not found'})
+
+    return jsonify(complete[room_code])
+
+
+@app.route('/list-rooms', methods=['GET'])
+def list_rooms():
+    room_info = {}
+    for room_code, player in rooms.items():
+        room_info[room_code] = len(player['players'])
+
+    return jsonify({'roomInfo': room_info})
 
 
 if __name__ == '__main__':