Desktop_AI_Agent/app.py at main · Anish-CodeDev/Desktop_AI_Agent · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
from agent import app as agent
from langchain_core.messages import HumanMessage
from PySide6.QtWidgets import QApplication, QWidget, QVBoxLayout, QTextEdit, QPushButton,QLineEdit,QHBoxLayout
import sys
from audio_record import record_speech
from gemini import transcript_audio
conversational_history = []

def run_agent(user_inp):
    global conversational_history
    conversational_history.append(HumanMessage(content=user_inp))
    result = agent.invoke({"messages":conversational_history})
    conversational_history = result['messages']
    return dict(conversational_history[-1])['content']


class ChatUI(QWidget):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("AI Agent")
        self.resize(500, 600)

        layout = QVBoxLayout()

        self.chat_area = QTextEdit()
        self.chat_area.setReadOnly(True)
        self.chat_area.setStyleSheet("""background-color: #343541;
                                        color: #D1D5DB;
                                        font-family: 'Arial';
                                        font-size: 14px;
                                        font-weight: bold;  """)
        layout.addWidget(self.chat_area)

        self.input_box = QLineEdit()
        self.input_box.setStyleSheet("""
                    background-color: #40414F;      /* Dark input background */
                    color: #FFFFFF;                 /* Typed text color */
                    font-family: 'Arial';
                    font-size: 14px;
                    border: 1px solid #6B7280;      /* Optional border */
                    border-radius: 5px;             /* Rounded corners */
                    padding: 5px;
                """)
        row = QHBoxLayout()
        self.input_box.setPlaceholderText("Type your question")
        row.addWidget(self.input_box)
        #layout.addWidget(self.input_box)
        self.mic_btn = QPushButton("🎤")
        self.mic_btn.clicked.connect(self.record_audio)
        self.mic_btn.setFixedWidth(40)
        self.mic_btn.setStyleSheet("""
            QPushButton {
                background-color: #FFFFFF;
                border-radius: 5px;
                font-size: 16px;
            }
            QPushButton:hover {
                background-color: #F0F0F0;
            }
        """)
        #layout.addWidget(self.mic_btn)
        row.addWidget(self.mic_btn)
        layout.addLayout(row)
        send_btn = QPushButton("Send")
        self.input_box.returnPressed.connect(self.send_message)
        send_btn.clicked.connect(self.send_message)
        send_btn.setStyleSheet("""
                QPushButton {
                    background-color: #FFFFFF;   /* White background */
                    color: #111111;              /* Dark text */
                    font-family: 'Arial';
                    font-size: 14px;
                    font-weight: bold;
                    border: 1px solid #CCCCCC;  /* Light gray border */
                    border-radius: 8px;          /* Rounded corners */
                    padding: 8px 16px;           /* Vertical and horizontal padding */
                }
                QPushButton:hover {
                    background-color: #F0F0F0;   /* Slightly gray on hover */
                }
                QPushButton:pressed {
                    background-color: #E0E0E0;   /* Darker gray when pressed */
                }
            """)

        layout.addWidget(send_btn)
        self.setLayout(layout)

    def send_message(self):
        inp_text = self.input_box.text()
        self.input_box.clear()
        self.chat_area.append(f"You: {inp_text}")
        res = run_agent(inp_text)
        self.chat_area.append(f"AI: {res}")

    def record_audio(self):
        print('Recording started')
        status = record_speech()
        if status:
            inp_text = transcript_audio('recordings/recording.mp3')
            self.input_box.clear()
            self.chat_area.append(f"You: {inp_text}")
            res = run_agent(inp_text)
            self.chat_area.append(f"AI: {res}")

        else:
            self.chat_area.append("AI: Could'nt hear anything from you")


if __name__ == '__main__':
    app = QApplication(sys.argv)
    window = ChatUI()
    window.setStyleSheet("""
    background-color: #343541;       /* Dark chat background */
    border-radius: 10px;             /* Rounded corners for modern look */
""")
    window.show()
    sys.exit(app.exec())