app.py

from json import JSONEncoder
from bson import ObjectId
from flask import Flask, request, jsonify
import os
import re
from flask_cors import CORS
from pymongo import MongoClient
from dotenv import load_dotenv
from pymongo.server_api import ServerApi
from datetime import datetime

from model.room import Room
from model.user import User
from widget.objectIdencoder import ObjectIdEncoder


app = Flask(__name__)
CORS(app)
load_dotenv()
# MongoDB connection
mongo_uri = os.getenv('MONGODB_URI')
database_name = os.getenv('DATABASE_NAME')
collection_name = os.getenv('COLLECTION_NAME')


client = MongoClient(mongo_uri, server_api=ServerApi('1'))
db = client[database_name]
collection = db[collection_name]
collection2 = db['users']
rooms = db['rooms']


# Problem statement search 
def extract_keywords(text):
    stopwords = {'and', 'the', 'is', 'of', 'or', 'in', 'to', 'it', 'that', 'was', 'with', 'for', 'on', 'as', 'by', 'at', 'an', 'but', 'not', 'are', 'you', 'we', 'can', 'have', 'has', 'this'}
    keywords = re.findall(r'\b\w+\b', text.lower())
    keywords = [keyword for keyword in keywords if keyword not in stopwords]
    return keywords

def calculate_percentage_match(input_keywords, statement_keywords):
    if len(input_keywords) == 0:
        return 0.0
    match_percentage = (len(set(input_keywords) & set(statement_keywords)) / len(input_keywords)) * 100
    return round(match_percentage, 2)

def search_problem_statements(input_text, min_percentage=20, page=1, results_per_page=10):
    input_keywords = extract_keywords(input_text)
    matching_statements = []

    for doc in collection.find():
        author = doc.get('Author', '')
        title = doc.get('Title', '')
        problem_statement = doc.get('Problem Statement', '')
        contributor = doc.get('Contributor', '')

        statement = f"{author} - {title} - {problem_statement} - {contributor}"
        statement_keywords = extract_keywords(statement)

        match_percentage = calculate_percentage_match(input_keywords, statement_keywords)
        if match_percentage > min_percentage:
            matching_statements.append({
                'percentage': match_percentage,
                'title': title,
                'author': author,
                'problem_statement': problem_statement,
                'contributor': contributor if contributor else ''
            })

    matching_statements.sort(key=lambda x: x['percentage'], reverse=True)
    return matching_statements



# login signup
def get_user_data(username):
    user_data = collection2.find_one({identify_input_type(username): username})
    return user_data


def check_user_exists(username):
    return get_user_data(username) is not None


def check_user_credentials(username, password):

    user_data = get_user_data(username)

    if user_data:
        if user_data['password'] == password:
            return True
        return False

def identify_input_type(input_str):
    if re.match(r'^[\w\.-]+@[\w\.-]+$', input_str):
        return 'email'
    else:
        return 'username'



# routess
@app.route('/register', methods = ['POST'])
def register():
    data = request.json
    name = data.get('name','')
    password = data.get('password', '')
    email = data.get('email', '')
    username = data.get('username','')
    gitlink = data.get('gitlink','')





    if not name or not password or not email:
        return jsonify({'status':'error', 'message': 'Username and password are required'})
    if check_user_exists(username):
        return jsonify({'status': 'error', 'message': 'Username already exist'})
    
    result = collection2.insert_one({
        'username': username,
        'password': password,
        'email':email,
        'gitlink': gitlink,
         'name': name           
                    })

    if result.inserted_id:
        return jsonify({'status': 'success', 'message':'User created successfully'})
    else:
        return jsonify({'status': 'error', 'message': 'Failed to create user'})
    



@app.route('/login', methods = ['POST'])
def login():
    data = request.json
    username = data.get('username', '')
    password = data.get('password', '')

    if not username or not password:
        return jsonify({'status': 'error', 'message': 'Username and password are required'})
    
    if check_user_credentials(username, password):
        user_data = collection2.find_one({identify_input_type(username): username}) 
        res = {'id': str(user_data['_id']),
                    'name': user_data['name'],
                    'username': user_data['username'],
                    'email': user_data['email'],
                    'gitlink': user_data['gitlink']
                    }
        return jsonify({'status': 'success', 'info':res})
    else:
        return jsonify({'status':'error', 'message':'Invalid name or password'})


@app.route('/search', methods=['POST'])
def search():
    try:
        # Check if MongoDB is connected
        client.server_info()
        print("Connected")
        # If connected, start the Flask app
        # app.run(debug=True, ssl_context='adhoc')
    except Exception as e:
        print("Failed to connect to MongoDB:", e)
    data = request.json
    input_text = data.get('input_text', '')
    min_percentage = data.get('min_percentage', 20)
    page = data.get('page', 1)

    results = search_problem_statements(input_text, min_percentage, page)

    if results:
        for i, statement in enumerate(results, start=1):
            print(f"{i}. Title: {statement['title']}, Author: {statement['author']}, Problem Statement: {statement['problem_statement']}, Contributor: {statement['contributor']}")

        response = {'status': 'success', 'results': results}
    else:
        response = {'status': 'error', 'message': 'No matching problem statements found.'}

    return jsonify(response)


@app.route('/create-room', methods=['POST'])
def create_room():
    data = request.json

    room = Room(data)
    if not room.room_name:
        return jsonify({"eNTER room_name"}),400
    if not room.team_members:
        return jsonify({"enter room_members"}),400
    if not room.problem_id:
        return jsonify({"enter problem_id"}),400

    mapp = {
        "room_id": room.room_id,
        "problem_id":str(room.problem_id),  
        "room_name": room.room_name,
        "team_members": room.team_members,
        "room_token": room.room_token,
        "videocon": room.videocon,
    }
    collect = rooms.insert_one(mapp)
    
    if collect.inserted_id:
        mapp['_id'] = str(mapp['_id'])
        return jsonify({"status":"success", "info": mapp })
    else:
        return jsonify({"status":"error", "message": "error occured"})
    


if __name__ == '__main__':
    try:
        # Check if MongoDB is connected
        client.server_info()
        print("Connected")
        # If connected, start the Flask app
        app.run(debug=True, ssl_context='adhoc')
    except Exception as e:
        print("Failed to connect to MongoDB:", e)