增加登陆日志留存模块

@@ -5,7 +5,11 @@ a = Analysis(
     ['image_generator.py'],
     ['image_generator.py'],
     pathex=[],
     pathex=[],
     binaries=[],
     binaries=[],
-    datas=[('config.json', '.')],
+    datas=[
+        ('config.json', '.'),
+        ('zb100_windows.ico', '.'),
+        ('zb100_mac.icns', '.')
+    ],
     hiddenimports=[],
     hiddenimports=[],
     hookspath=[],
     hookspath=[],
     hooksconfig={},
     hooksconfig={},

@@ -95,9 +95,32 @@ class LoginDialog(QDialog):
         self.password_changed = False
         self.password_changed = False
         self.current_password_hash = ""
         self.current_password_hash = ""
 
 
+        self.set_window_icon()
         self.setup_ui()
         self.setup_ui()
         self.apply_styles()
         self.apply_styles()
 
 
+    def set_window_icon(self):
+        """Set window icon based on platform"""
+        if getattr(sys, 'frozen', False):
+            # Running as compiled executable
+            if platform.system() == 'Windows':
+                icon_path = os.path.join(sys._MEIPASS, 'zb100_windows.ico')
+            elif platform.system() == 'Darwin':
+                icon_path = os.path.join(sys._MEIPASS, 'zb100_mac.icns')
+            else:
+                icon_path = None
+        else:
+            # Running as script
+            if platform.system() == 'Windows':
+                icon_path = 'zb100_windows.ico'
+            elif platform.system() == 'Darwin':
+                icon_path = 'zb100_mac.icns'
+            else:
+                icon_path = None
+
+        if icon_path and os.path.exists(icon_path):
+            self.setWindowIcon(QIcon(icon_path))
+
     def setup_ui(self):
     def setup_ui(self):
         """Build login dialog UI"""
         """Build login dialog UI"""
         self.setWindowTitle("登录 - AI 图像生成器")
         self.setWindowTitle("登录 - AI 图像生成器")
@@ -434,9 +457,32 @@ class ImageGeneratorWindow(QMainWindow):
         self.saved_password_hash = ""
         self.saved_password_hash = ""
 
 
         self.load_config()
         self.load_config()
+        self.set_window_icon()
         self.setup_ui()
         self.setup_ui()
         self.apply_styles()
         self.apply_styles()
 
 
+    def set_window_icon(self):
+        """Set window icon based on platform"""
+        if getattr(sys, 'frozen', False):
+            # Running as compiled executable
+            if platform.system() == 'Windows':
+                icon_path = os.path.join(sys._MEIPASS, 'zb100_windows.ico')
+            elif platform.system() == 'Darwin':
+                icon_path = os.path.join(sys._MEIPASS, 'zb100_mac.icns')
+            else:
+                icon_path = None
+        else:
+            # Running as script
+            if platform.system() == 'Windows':
+                icon_path = 'zb100_windows.ico'
+            elif platform.system() == 'Darwin':
+                icon_path = 'zb100_mac.icns'
+            else:
+                icon_path = None
+
+        if icon_path and os.path.exists(icon_path):
+            self.setWindowIcon(QIcon(icon_path))
+
     def get_config_dir(self):
     def get_config_dir(self):
         """Get the appropriate directory for config files based on platform and mode"""
         """Get the appropriate directory for config files based on platform and mode"""
         if getattr(sys, 'frozen', False):
         if getattr(sys, 'frozen', False):
@@ -555,7 +601,7 @@ class ImageGeneratorWindow(QMainWindow):
         self.img_scroll.setWidgetResizable(True)
         self.img_scroll.setWidgetResizable(True)
         self.img_scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
         self.img_scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
         self.img_scroll.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
         self.img_scroll.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
-        self.img_scroll.setFixedHeight(140)
+        self.img_scroll.setFixedHeight(160)
 
 
         self.img_container = QWidget()
         self.img_container = QWidget()
         self.img_layout = QHBoxLayout()
         self.img_layout = QHBoxLayout()
@@ -1118,6 +1164,28 @@ def main():
     # Create QApplication
     # Create QApplication
     app = QApplication(sys.argv)
     app = QApplication(sys.argv)
 
 
+    # Set application icon
+    if getattr(sys, 'frozen', False):
+        # Running as compiled executable
+        if platform.system() == 'Windows':
+            icon_path = os.path.join(sys._MEIPASS, 'zb100_windows.ico')
+        elif platform.system() == 'Darwin':
+            icon_path = os.path.join(sys._MEIPASS, 'zb100_mac.icns')
+        else:
+            icon_path = None
+    else:
+        # Running as script
+        if platform.system() == 'Windows':
+            icon_path = 'zb100_windows.ico'
+        elif platform.system() == 'Darwin':
+            icon_path = 'zb100_mac.icns'
+        else:
+            icon_path = None
+
+    if icon_path and os.path.exists(icon_path):
+        app_icon = QIcon(icon_path)
+        app.setWindowIcon(app_icon)
+
     # Check database config
     # Check database config
     if not db_config:
     if not db_config:
         QMessageBox.critical(None, "配置错误",
         QMessageBox.critical(None, "配置错误",