Update client_chat.py

client_chat.py

@@ -3,17 +3,17 @@ import requests
 import json
 from PyQt6.QtWidgets import (
     QApplication, QWidget, QVBoxLayout, QHBoxLayout,
-    QPushButton, QTextEdit, QLineEdit, QLabel, QComboBox, QScrollArea
+    QPushButton, QTextEdit, QLineEdit, QLabel, QComboBox
 )
 from PyQt6.QtCore import QTimer, Qt
-from PyQt6.QtGui import QFont, QCursor, QIcon
+from PyQt6.QtGui import QFont, QCursor, QIcon, QColor, QTextCursor
 
 API_BASE = "http://127.0.0.1:7860/api"
 
 class ChatClient(QWidget):
-    def __init__(self):
+    def __init__(self, token):
         super().__init__()
-        self.token = None
+        self.token = token  # Token comes from main.py login
         self.ai_typing_timer = None
         self.ai_text_buffer = ""
         self.setWindowTitle("AI HUB Desktop Client")
@@ -21,41 +21,29 @@ class ChatClient(QWidget):
 
         self.init_ui()
         self.showMaximized()
-        self.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))
+        self.setCursor(QCursor(Qt.CursorShape.ArrowCursor))
 
     def init_ui(self):
         self.layout = QVBoxLayout(self)
         self.setLayout(self.layout)
 
-        # Login area
-        login_layout = QHBoxLayout()
-        self.email_input = QLineEdit()
-        self.email_input.setPlaceholderText("Email")
-        login_layout.addWidget(self.email_input)
-
-        self.password_input = QLineEdit()
-        self.password_input.setPlaceholderText("Password")
-        self.password_input.setEchoMode(QLineEdit.EchoMode.Password)
-        login_layout.addWidget(self.password_input)
-
-        self.login_button = QPushButton("Login")
-        self.login_button.clicked.connect(self.handle_login)
-        login_layout.addWidget(self.login_button)
-
-        self.signup_button = QPushButton("Signup")
-        self.signup_button.clicked.connect(self.handle_signup)
-        login_layout.addWidget(self.signup_button)
-        self.layout.addLayout(login_layout)
-
         # Chat Display Area
         self.chat_area = QTextEdit()
         self.chat_area.setReadOnly(True)
-        self.chat_area.setFont(QFont("Arial", 12))
+        self.chat_area.setFont(QFont("Consolas", 12))
+        self.chat_area.setStyleSheet("""
+            QTextEdit {
+                background-color: #1e1e1e;
+                color: #f1f1f1;
+                border-radius: 10px;
+                padding: 10px;
+            }
+        """)
         self.layout.addWidget(self.chat_area)
 
         # Typing Indicator
         self.typing_label = QLabel("")
-        self.typing_label.setStyleSheet("color: green; font-style: italic;")
+        self.typing_label.setStyleSheet("color: #00ff00; font-style: italic;")
         self.layout.addWidget(self.typing_label)
 
         # Input Area
@@ -63,15 +51,40 @@ class ChatClient(QWidget):
         self.user_input = QLineEdit()
         self.user_input.setPlaceholderText("Type your message here...")
         self.user_input.returnPressed.connect(self.send_message)
+        self.user_input.setStyleSheet("""
+            QLineEdit {
+                border-radius: 8px;
+                padding: 6px;
+                font-size: 14px;
+            }
+        """)
         input_layout.addWidget(self.user_input)
 
         self.send_button = QPushButton("Send")
+        self.send_button.setStyleSheet("""
+            QPushButton {
+                background-color: #0078d7;
+                color: white;
+                border-radius: 8px;
+                padding: 6px 12px;
+            }
+            QPushButton:hover {
+                background-color: #005a9e;
+            }
+        """)
         self.send_button.clicked.connect(self.send_message)
         input_layout.addWidget(self.send_button)
 
         # Model selection
         self.model_select = QComboBox()
         self.model_select.addItems(["qwen", "deepseek"])
+        self.model_select.setStyleSheet("""
+            QComboBox {
+                border-radius: 8px;
+                padding: 4px;
+                font-size: 14px;
+            }
+        """)
         input_layout.addWidget(self.model_select)
 
         self.layout.addLayout(input_layout)
@@ -79,7 +92,7 @@ class ChatClient(QWidget):
     # --------------------
     # Display AI typing animation
     # --------------------
-    def type_ai_message(self, full_text, delay=50):
+    def type_ai_message(self, full_text, delay=30):
         self.ai_text_buffer = full_text
         self.typing_label.setText("🤖 AI is typing...")
         self.ai_typing_timer = QTimer()
@@ -89,9 +102,9 @@ class ChatClient(QWidget):
     def _type_next_char(self):
         if self.ai_text_buffer:
             char = self.ai_text_buffer[0]
+            self.chat_area.moveCursor(QTextCursor.MoveOperation.End)
             self.chat_area.insertPlainText(char)
             self.ai_text_buffer = self.ai_text_buffer[1:]
-            self.chat_area.moveCursor(Qt.TextCursor.MoveOperation.End)
         else:
             self.ai_typing_timer.stop()
             self.typing_label.setText("")
@@ -101,7 +114,7 @@ class ChatClient(QWidget):
     # --------------------
     def send_message(self):
         message = self.user_input.text().strip()
-        if not message or not self.token:
+        if not message:
             return
         self.chat_area.append(f"🧑 {message}")
         self.user_input.clear()
@@ -126,58 +139,13 @@ class ChatClient(QWidget):
         except Exception as e:
             self.chat_area.append(f"Error sending message: {e}")
 
-    # --------------------
-    # Login / Signup
-    # --------------------
-    def handle_login(self):
-        email = self.email_input.text().strip()
-        password = self.password_input.text().strip()
-        if self.login(email, password):
-            self.chat_area.append("✅ Logged in successfully!")
-        else:
-            self.chat_area.append("❌ Login failed!")
-
-    def handle_signup(self):
-        email = self.email_input.text().strip()
-        password = self.password_input.text().strip()
-        if not email or not password:
-            self.chat_area.append("⚠️ Enter email & password")
-            return
-        try:
-            res = requests.post(
-                f"{API_BASE}/signup",
-                headers={"Content-Type": "application/json"},
-                data=json.dumps({"email": email, "username": email.split('@')[0], "password": password})
-            )
-            data = res.json()
-            if "message" in data:
-                self.chat_area.append("✅ Signup successful! Verify your email.")
-            else:
-                self.chat_area.append("❌ Signup error: " + data.get("error", "Unknown"))
-        except Exception as e:
-            self.chat_area.append(f"Error signing up: {e}")
-
-    def login(self, email, password):
-        try:
-            res = requests.post(
-                f"{API_BASE}/login",
-                headers={"Content-Type": "application/json"},
-                data=json.dumps({"email": email, "password": password})
-            )
-            data = res.json()
-            if "token" in data:
-                self.token = data["token"]
-                return True
-            return False
-        except Exception as e:
-            print("Login exception:", e)
-            return False
-
 # --------------------
 # Run App
 # --------------------
 if __name__ == "__main__":
+    # Example: Pass token from main.py login
+    token = "YOUR_JWT_TOKEN_HERE"
     app = QApplication(sys.argv)
-    window = ChatClient()
+    window = ChatClient(token)
     window.show()
     sys.exit(app.exec())