from PyQt6.QtGui import QCloseEvent
from PyQt6.QtWidgets import QApplication, QMainWindow, QTextEdit, QVBoxLayout, QHBoxLayout, QPushButton, QFileDialog, QWidget, QSizePolicy
from PyQt6.QtCore import pyqtSignal
import sys
import os
from pydub import AudioSegment
from pydub.playback import play

class FileViewerApp(QMainWindow):
    
    
    window_closed = pyqtSignal()

    def __init__(self):
        super().__init__()

        

    def init_ui(self, title):
        # Create widgets
        self.text_edit = QTextEdit(self)
        self.text_edit.setReadOnly(True)

        self.close_button = QPushButton('Close', self)
        self.close_button.setSizePolicy(QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Fixed)
        self.close_button.setFixedWidth(80)  # Set width to 60 pixels
        self.close_button.clicked.connect(self.close)

        self.listen_button = QPushButton('Listen', self)
        self.listen_button.setSizePolicy(QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Fixed)
        self.listen_button.setFixedWidth(80)  # Set width to 60 pixels
        self.listen_button.clicked.connect(self.listen)

        # Set up layout
        layout = QVBoxLayout()
        
        layout.addWidget(self.text_edit)

        # Widget to hold buttons
        button_widget = QWidget(self)
        button_layout = QHBoxLayout(button_widget)
        button_layout.addStretch(1)  # Add stretchable space before buttons
        button_layout.addWidget(self.close_button)
        button_layout.addWidget(self.listen_button)
        layout.addWidget(button_widget)

        central_widget = QWidget()
        central_widget.setLayout(layout)
        self.setCentralWidget(central_widget)

        # Set window properties
        self.setWindowTitle(title)
        self.setGeometry(100, 100, 800, 600)

        # Apply styles
        self.setStyleSheet("""
            QMainWindow {
                background-color: #f0f0f0;
            }
            QTextEdit {
                background-color: white;
                border: 1px solid #ddd;
                padding: 8px;
                font-size: 14px;
                color: #333;
            }
            QPushButton {
                background-color: #4CAF50;
                color: white;
                padding: 8px 16px;
                font-size: 18px;
                border: none;
                border-radius: 4px;
            }
            QPushButton:hover {
                background-color: #45a049;
            }
        """)


    def closeEvent(self, event):
        self.window_closed.emit()
        event.accept()

    def listen(self, audio_file):
        audio_file = r'C:\Users\Shiv\dev\InfoVeiwer-Final_Year_Project-\received\sample.mp3'
        if os.path.exists(audio_file):
            try:
                # Load the audio file
                audio = AudioSegment.from_file(audio_file)
                # Play the audio
                play(audio)
            except Exception as e:
                print("Error:", e)
                import traceback
                traceback.print_exc()  # Print the stack trace for detailed debugging
        else:
            print("File not found:", audio_file)

    def load_file(self, file_path):
        # Read the content of the selected file
        with open(file_path, 'r') as file:
            file_content = file.read()
        # Display the content in the QTextEdit widget
        self.text_edit.setPlainText(file_content)

    
def main():
    app = QApplication(sys.argv)
    main_win = FileViewerApp()
    main_win.show()
    main_win.init_ui(title="Sample text")
    main_win.load_file(r'C:\Users\Shiv\dev\InfoVeiwer-Final_Year_Project-\received\sample.txt')
    sys.exit(app.exec())



if __name__ == '__main__':
    main()