main.py

import datetime
import flet as ft
from signin_form import *
from signup_form import *
from chat_message import *
import predict
import complaint
import firebase_admin
from firebase_admin import credentials
from firebase_admin import firestore
import hashlib
cred = credentials.Certificate(
    "your_firebase_API_key.json")
firebase_admin.initialize_app(cred)

print("Server in ONLINE!")

db = firestore.client()


def main(page: ft.Page):
    def sha256_hash(input_string):

        sha256 = hashlib.sha256()

        sha256.update(input_string.encode('utf-8'))

        hashed_string = sha256.hexdigest()

        return hashed_string
    page.title = "Safe Messenger"
    page.vertical_alignment = ft.MainAxisAlignment.CENTER
    page.horizontal_alignment = ft.CrossAxisAlignment.CENTER

    def dropdown_changed(e):

        new_message.value = new_message.value + emoji_list.value
        page.update()

    def close_banner(e):

        page.banner.open = False
        page.update()

    def open_dlg():
        page.dialog = dlg
        dlg.open = True
        page.update()

    def close_dlg(e):
        dlg.open = False
        page.route = "/"
        page.update()

    def open_aff():
        page.dialog = aff
        aff.open = True
        page.update()

    def close_aff(e):
        aff.open = False
        page.route = "/"
        page.update()

    def req_unban(user, password):
        users_ref = db.collection("users").document(user)
        docs = users_ref.get()
        data = docs.to_dict()
        if data["password"] != sha256_hash(password):
            ShowBanner("Password Incorrect")
        elif not docs.exists:
            ShowBanner("User Not Found")
        elif not data["is_banned"]:
            ShowBanner("User is not banned")
        else:
            users_ref.update({"raised": True})
            open_aff()

    def sign_in(user: str, password: str):
        users_ref = db.collection("users").document(user)
        docs = users_ref.get()
        auth = False
        ban = False
        msg = 'No user with the Username Found!'
        if docs.exists:
            auth = True
            msg = ''
            data = docs.to_dict()
            if data["password"] != sha256_hash(password):
                auth = False
                msg = "Log in failed, Incorrect User Name or Password"
            if data["is_banned"]:
                ban = True
                msg = "You have been Banned from the chat room!"
        if not auth or ban:
            ShowBanner(msg)
            print("LOG ERR")

        else:
            print("Redirecting to chat...")
            page.session.set("user", user)
            page.route = "/chat"
            page.pubsub.send_all(
                Message(
                    user=user,
                    text=f"{user} has joined the chat.",
                    message_type="login_message",
                )
            )
            page.update()

    def sign_up(user: str, password: str, mobile):

        doc_ref = db.collection("users").document(user)

        if doc_ref.set({"username": user, "password": sha256_hash(password), "mobile": mobile, "is_banned": False, "warnings": 0, "prev_bans": 0, "total_warnings": 0}):
            print("Successfully Registered User...")
            open_dlg()

    def on_message(message: Message):
        if message.message_type == "chat_message":
            m = ChatMessage(message)
        else:
            m = ft.Text(message.text, italic=True,
                        color=ft.colors.WHITE, size=12)

        chat.controls.append(m)
        page.update()

    page.pubsub.subscribe(on_message)

    def send_message_click(e):
        db = firestore.client()
        message = new_message.value
        prediction = predict.predict_text(message)
        user = page.session.get("user")
        users_ref = db.collection("users").document(user)
        docs = users_ref.get()
        data = docs.to_dict()
        # Track the number of warnings given to the user
        warning_count = data["warnings"]
        if prediction > 50:

            users_ref.update({"warnings": warning_count+1})

            # Show an alert message to the user if the warning count is greater than 0
            if warning_count < 3:
                page.pubsub.send_all(
                    Message(
                        user=page.session.get("user"),
                        text=f"The message doesn't follow the policy of the company.{user} has {3 - warning_count} warnings remaining.",
                        message_type="alert",
                    )
                )

            # If the warning count is equal to 3, generate a complaint document and remove the user from the chat room
            elif warning_count == 3:

                number = data["mobile"]
                complaint.generate_complaint_document(
                    page.session.get("user"), number, message, datetime.datetime.now())
                page.pubsub.send_all(
                    Message(
                        user=page.session.get("user"),
                        text=f"{user} is banned from the chat room due to incomplaince with company policy.Their future messages will not reflect in the chat room",
                        message_type="alert",
                    )
                )
                users_ref.update({"is_banned": True})
                users_ref.update({
                                 "prev_bans": data["prev_bans"]+1})
                users_ref.update({"message": message})
                page.session.remove("user")
                page.route = "/"
                page.update()
            else:
                page.pubsub.send_all(
                    Message(
                        user=page.session.get("user"),
                        text="This message is not displayed due to company policy",
                        message_type="alert",
                    )
                )

        else:
            if warning_count >= 3:
                page.pubsub.send_all(
                    Message(
                        user=page.session.get("user"),
                        text="This message is not displayed due to company policy",
                        message_type="alert",
                    )
                )
            else:
                page.pubsub.send_all(
                    Message(
                        user=page.session.get("user"),
                        text=message,
                        message_type="chat_message",
                    )
                )

            # Clear the message input field
            new_message.value = ""

            # Update the page
            page.update()

    def btn_signin(e):
        page.route = "/"
        page.update()

    def btn_signup(e):
        page.route = "/signup"
        page.update()

    def btn_exit(e):
        page.session.remove("user")
        page.route = "/"
        page.update()

    principal_content = ft.Column(
        [
            ft.Icon(ft.icons.WECHAT, size=200, color=ft.colors.BLUE),
            ft.Text(value="Safe Messenger",
                    size=50, color=ft.colors.BLACK),
        ],
        height=400,
        width=600,
        alignment=ft.MainAxisAlignment.CENTER,
        horizontal_alignment=ft.CrossAxisAlignment.CENTER,
    )
    emoji_list = ft.Dropdown(
        on_change=dropdown_changed,
        options=[
            ft.dropdown.Option("😃"),
            ft.dropdown.Option("😊"),
            ft.dropdown.Option("😂"),
            ft.dropdown.Option("🤔"),
            ft.dropdown.Option("😭"),
            ft.dropdown.Option("😉"),
            ft.dropdown.Option("🤩"),
            ft.dropdown.Option("🥰"),
            ft.dropdown.Option("😎"),
            ft.dropdown.Option("❤️"),
            ft.dropdown.Option("🔥"),
            ft.dropdown.Option("✅"),
            ft.dropdown.Option("✨"),
            ft.dropdown.Option("👍"),
            ft.dropdown.Option("🎉"),
            ft.dropdown.Option("👉"),
            ft.dropdown.Option("⭐"),
            ft.dropdown.Option("☀️"),
            ft.dropdown.Option("👀"),
            ft.dropdown.Option("👇"),
            ft.dropdown.Option("🚀"),
            ft.dropdown.Option("🎂"),
            ft.dropdown.Option("💕"),
            ft.dropdown.Option("🏡"),
            ft.dropdown.Option("🍎"),
            ft.dropdown.Option("🎁"),
            ft.dropdown.Option("💯"),
            ft.dropdown.Option("💤"),
        ],
        width=50,
        value="😃",
        alignment=ft.alignment.center,
        border_color=ft.colors.AMBER,
        color=ft.colors.AMBER,
    )

    signin_UI = SignInForm(sign_in, req_unban, btn_signup)
    signup_UI = SignUpForm(sign_up, btn_signin)

    chat = ft.ListView(
        expand=True,
        spacing=10,
        auto_scroll=True,
    )

    new_message = ft.TextField(
        hint_text="Write a message...",
        autofocus=True,
        shift_enter=True,
        min_lines=1,
        max_lines=5,
        filled=True,
        expand=True,
        on_submit=send_message_click,
    )

    def ShowBanner(msg):
        page.banner = ft.Banner(
            bgcolor=ft.colors.BLACK45,
            leading=ft.Icon(ft.icons.ERROR, color=ft.colors.RED, size=40),
            content=ft.Text(msg),
            actions=[
                ft.TextButton("Ok", on_click=close_banner),
            ],
        )
        page.banner.open = True
        page.update()

    dlg = ft.AlertDialog(
        modal=True,
        title=ft.Container(
            content=ft.Icon(
                name=ft.icons.CHECK_CIRCLE_OUTLINED, color=ft.colors.GREEN, size=100
            ),
            width=120,
            height=120,
        ),
        content=ft.Text(
            value="Congratulations,\n your account has been successfully created\n Please Sign In",
            text_align=ft.TextAlign.CENTER,
        ),
        actions=[
            ft.ElevatedButton(
                text="Continue", color=ft.colors.WHITE, on_click=close_dlg
            )
        ],
        actions_alignment="center",
        on_dismiss=lambda e: print("Dialog dismissed!"),
    )
    aff = ft.AlertDialog(
        modal=True,
        title=ft.Container(
            content=ft.Icon(
                name=ft.icons.CHECK_CIRCLE_OUTLINED, color=ft.colors.GREEN, size=100
            ),
            width=120,
            height=120,
        ),
        content=ft.Text(
            value="Request Raised!",
            text_align=ft.TextAlign.CENTER,
        ),
        actions=[
            ft.ElevatedButton(
                text="Continue", color=ft.colors.WHITE, on_click=close_aff
            )
        ],
        actions_alignment="center",
        on_dismiss=lambda e: print("Dialog dismissed!"),
    )

    def route_change(route):
        if page.route == "/admin":
            admin.initialize()
        if page.route == "/":
            page.clean()
            page.add(
                ft.Row(
                    [principal_content, signin_UI],
                    alignment=ft.MainAxisAlignment.CENTER,
                )
            )

        if page.route == "/signup":
            page.clean()
            page.add(
                ft.Row(
                    [principal_content, signup_UI],
                    alignment=ft.MainAxisAlignment.CENTER,
                )
            )

        if page.route == "/chat":
            if page.session.contains_key("user"):
                page.clean()
                page.add(
                    ft.Row(
                        [
                            ft.Text(value="Safe Messenger",
                                    color=ft.colors.WHITE),
                            ft.ElevatedButton(
                                text="Log Out",
                                bgcolor=ft.colors.RED_800,
                                on_click=btn_exit,
                            ),
                        ],
                        alignment=ft.MainAxisAlignment.SPACE_AROUND,
                    )
                )
                page.add(
                    ft.Container(
                        content=chat,
                        border=ft.border.all(1, ft.colors.OUTLINE),
                        border_radius=5,
                        padding=10,
                        expand=True,
                    )
                )
                page.add(
                    ft.Row(
                        controls=[
                            emoji_list,
                            new_message,
                            ft.IconButton(
                                icon=ft.icons.SEND_ROUNDED,
                                tooltip="Send message",
                                on_click=send_message_click,
                            ),
                        ],
                    )
                )

            else:
                page.route = "/"
                page.update()

    page.on_route_change = route_change
    page.add(
        ft.Row([principal_content, signin_UI],
               alignment=ft.MainAxisAlignment.CENTER)
    )


ft.app(target=main, view=ft.WEB_BROWSER, )