import os

import threading

#os.environ["KIVY_NO_CONSOLELOG"] = "1"

from kivy.lang import Builder

from kivymd.app import MDApp

from kivymd.uix.boxlayout import MDBoxLayout

from kivymd.uix.button import MDRaisedButton

from kivymd.uix.label import MDLabel

from kivy.uix.image import Image

from kivy.graphics.texture import Texture

from kivy.clock import Clock

from kivy.uix.progressbar import ProgressBar

from kivy.uix.popup import Popup

from internal.frame_process import model_initialization, on_update

from internal.tts.tttest import generate_audios, play_audio

from kivy.core.text import Label

from kivy.graphics import Line, Rectangle, Color

from kivy.uix.widget import Widget

from kivy.clock import Clock

from vendor.kvconfigs import KV

import cv2

from notifypy import Notify

# Set the application icon

notification = Notify()

class AutopostureApp(MDApp):

    def build(self):

        generate_audios('bad', 'Sit correctly!')

        self.recently_alerted = False

        self.theme_cls.theme_style_switch_animation = True

        self.theme_cls.theme_style = "Light"

        self.theme_cls.primary_palette = "Blue"

        self.icon = 'assets/ergonomic.png'

        self.prediction_scores = []  # List to store prediction scores

        return Builder.load_string(KV)

    def on_start(self):

        self.video_running = False

        self.capture = cv2.VideoCapture(0, cv2.CAP_V4L2)

        model_initialization('0', 'src_models/yolov7-w6-pose.pt')

    def load_video_button(self):

        if not self.video_running:

            self.root.ids.start_autoposture.text = "PAUSE AUTOPOSTURE"

            self.video_running = True

            Clock.schedule_interval(self.load_video, 1.0 / 30.0)

        else:

            self.root.ids.start_autoposture.text = "RESTART AUTOPOSTURE"

            self.video_running = False

            Clock.unschedule(self.load_video)

    def switch_theme_style(self):

        self.theme_cls.primary_palette = (

            "Blue" if self.theme_cls.primary_palette == "Blue" else "Blue"

        )

        self.theme_cls.theme_style = (

            "Dark" if self.theme_cls.theme_style == "Light" else "Light"

        )

    def alert_user(self):

        notification.title = "Bad Posture"

        notification.message = "You are in a bad posture. Please correct your sitting."

        notification.icon = "assets/ergonomic.png"

        notification.send()

        play_audio('bad')

    def run_alert(self):

        if not self.recently_alerted:

            self.recently_alerted = False

            threading.Thread(target=self.alert_user).start()

    def trigger_alert(self, dt):

        self.alert_user()

        self.recently_alerted = False

    def load_video(self, *args):

        if not self.video_running:

            return

        ret, frame = self.capture.read()

        frame, posture, prediction_score, should_update = on_update(frame, self.recently_alerted, 0.7)

        if should_update:

            self.run_alert()

            self.recently_alerted = True

        else:

            self.recently_alerted = False

        buffer = cv2.flip(frame, 0).tobytes()

        texture = Texture.create(

            size=(frame.shape[1], frame.shape[0]),

            colorfmt='bgr')

        texture.blit_buffer(buffer, colorfmt='bgr', bufferfmt='ubyte')

        video_image = self.root.ids.video_image

        video_image.texture = texture

        # Update posture label and set the text color

        posture_label = self.root.ids.posture_label

        posture_label.text = f"Posture: {posture}"

        posture_label.theme_text_color = "Error" if posture == "bad" else "Primary"

        prediction_label = self.root.ids.prediction_label

        prediction_label.text = f"Prediction Score: {int(prediction_score * 100)}%"

        self.prediction_scores.append(prediction_score)

    def show_average_popup(self):

        if self.prediction_scores:

            average = int(sum(self.prediction_scores) / len(self.prediction_scores) * 100)

            content = MDBoxLayout(orientation='vertical', padding=10)

            progress_label = MDLabel(

                text=f"Average Prediction Score: {average}%",

                theme_text_color="Secondary",

                font_style="H6",

            )

            content.add_widget(progress_label)

            popup = Popup(

                title="Average Score",

                content=content,

                size_hint=(None, None),

                size=("350dp", "200dp"),

            )

            # progress_label.theme_text_color = "Primary"

            if self.theme_cls.theme_style == "Light":

            # If the theme is "Light," set the background color and font color accordingly

                popup.background_color = self.theme_cls.bg_dark

            popup.open()

        # Clear the prediction scores list

            self.prediction_scores = []

        else:

        # Handle the case where there are no prediction scores.

            content = MDLabel(text="No prediction scores yet")

            popup = Popup(

                title="Average Score",

                content=content,

                size_hint=(None, None),

                size=("350dp", "200dp"),  # Adjust the size here to make it wider

            )

            if self.theme_cls.theme_style == "Light":

            # If the theme is "Light," set the background color and font color accordingly

                popup.background_color = self.theme_cls.bg_dark

                content.theme_text_color = "Primary"

            popup.open()

if __name__ == '__main__':

    AutopostureApp().run()