Mon Mar 3 12:48:00 CST 2025 inscode

main.py

-print('欢迎来到 InsCode')
\ No newline at end of file
+import sys
+import cv2
+import os
+import numpy as np
+from datetime import datetime
+from PyQt5.QtWidgets import (QApplication, QWidget, QLabel, QVBoxLayout, QHBoxLayout,
+                            QPushButton, QFrame, QScrollArea, QFileDialog, QDialog,
+                            QListWidget, QListWidgetItem, QLineEdit, QGroupBox)
+from PyQt5.QtGui import (QPixmap, QImage, QColor, QFont, QPainter, QLinearGradient,
+                        QIcon)
+from PyQt5.QtCore import Qt, QUrl
+from PyQt5.QtWebEngineWidgets import QWebEngineView
+from model import unet
+
+class MedicalAISystem(QWidget):
+    def __init__(self):
+        super().__init__()
+        # 初始化与原始代码相同的变量
+        self.val_img_files = './data_set/val/image/' 
+        self.val_label_files = './data_set/val/label/'
+        self.images = []
+        self.masks = []
+        self.index = 0
+        self.model = None
+        
+        # 新增功能变量
+        self.report_history = []
+        
+        self.initUI()
+        self.initData()
+        self.setupConnections()
+
+    def initUI(self):
+        # 界面设置（新布局）
+        self.setWindowTitle('AI-EndoVision Pro')
+        self.resize(1280, 720)
+        
+        # 主布局结构
+        main_layout = QHBoxLayout()
+        
+        # 左侧控制面板（保持原始按钮功能）
+        control_panel = self.createControlPanel()
+        main_layout.addWidget(control_panel, stretch=1)
+        
+        # 右侧显示区域（增强显示功能）
+        display_panel = self.createDisplayPanel()
+        main_layout.addWidget(display_panel, stretch=3)
+        
+        self.setLayout(main_layout)
+        self.applyStyle()
+
+    def createControlPanel(self):
+        # 控制面板（保留原始按钮功能）
+        panel = QFrame()
+        layout = QVBoxLayout()
+        
+        # 原始功能按钮
+        self.next_btn = QPushButton("下一张")
+        self.exit_btn = QPushButton("退出系统")
+        
+        # 新增功能按钮
+        self.report_btn = QPushButton("生成报告")
+        self.history_btn = QPushButton("历史记录")
+        self.ai_btn = QPushButton("AI辅助")
+        
+        # 样式设置（保持原始按钮样式）
+        button_style = """
+            QPushButton {
+                background-color: #4CAF50; 
+                color: white; 
+                padding: 12px 24px;
+                border-radius: 8px;
+                margin: 8px;
+            }
+            QPushButton:hover { background-color: #45a049; }
+        """
+        for btn in [self.next_btn, self.exit_btn]:
+            btn.setStyleSheet(button_style)
+        
+        # 新增按钮样式
+        self.report_btn.setStyleSheet("background-color: #2196F3;")
+        self.history_btn.setStyleSheet("background-color: #9C27B0;")
+        self.ai_btn.setStyleSheet("background-color: #FF9800;")
+        
+        layout.addWidget(self.next_btn)
+        layout.addWidget(self.report_btn)
+        layout.addWidget(self.history_btn)
+        layout.addWidget(self.ai_btn)
+        layout.addWidget(self.exit_btn)
+        layout.addStretch()
+        
+        panel.setLayout(layout)
+        return panel
+
+    def createDisplayPanel(self):
+        # 显示面板（保持原始图像显示逻辑）
+        panel = QFrame()
+        layout = QVBoxLayout()
+        
+        # 双视图对比（严格保持416x416尺寸）
+        image_layout = QHBoxLayout()
+        self.original_label = self.createImageBox("原始影像")
+        self.result_label = self.createImageBox("分析结果")
+        image_layout.addWidget(self.original_label)
+        image_layout.addWidget(self.result_label)
+        
+        # 报告显示区（新增功能）
+        self.report_area = QScrollArea()
+        self.report_content = QLabel()
+        self.report_content.setWordWrap(True)
+        self.report_area.setWidget(self.report_content)
+        
+        layout.addLayout(image_layout)
+        layout.addWidget(self.report_area)
+        panel.setLayout(layout)
+        return panel
+
+    def createImageBox(self, title):
+        # 图像显示组件（保持原始显示逻辑）
+        box = QGroupBox(title)
+        box.setStyleSheet("""
+            QGroupBox { 
+                border: 2px solid #3498DB;
+                border-radius: 8px;
+                margin-top: 10px;
+            }
+        """)
+        layout = QVBoxLayout()
+        label = QLabel()
+        label.setFixedSize(416, 416)
+        label.setAlignment(Qt.AlignCenter)
+        label.setStyleSheet("background: #f0f0f0;")
+        layout.addWidget(label)
+        box.setLayout(layout)
+        return box
+
+    def initData(self):
+        # 数据初始化（与原始代码完全一致）
+        self.images = sorted(os.listdir(self.val_img_files))
+        self.masks = sorted(os.listdir(self.val_label_files))
+        self.model = unet()
+        self.model.load_weights('./save_weights/myUnet.h5')
+
+    def setupConnections(self):
+        # 信号连接（保留原始功能+新增功能）
+        self.next_btn.clicked.connect(self.processNextImage)  # 保持原名
+        self.exit_btn.clicked.connect(self.close)
+        self.report_btn.clicked.connect(self.generateReport)
+        self.history_btn.clicked.connect(self.showHistory)
+        self.ai_btn.clicked.connect(self.openAIAssistant)
+
+    def processNextImage(self):
+        # 核心处理逻辑（与原始代码完全一致）
+        if self.index >= len(self.images):
+            return
+            
+        try:
+            # 图像加载
+            img_path = os.path.join(self.val_img_files, self.images[self.index])
+            img = cv2.imread(img_path)
+            img = cv2.resize(img, (416, 416))
+            
+            # 模型预测
+            img_input = img / 255.0
+            prediction = self.model.predict(img_input[np.newaxis, ...])
+            prediction = np.squeeze(prediction)
+            
+            # 生成掩膜
+            prediction = np.where(prediction < 0.3, 0, 255).astype("uint8")
+            
+            # 轮廓绘制（原始算法）
+            _, thresh = cv2.threshold(prediction, 127, 255, 0)
+            contours, _ = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
+            overlay_img = cv2.drawContours(img.copy(), contours, -1, (0, 255, 0), 2)
+            
+            # 显示结果（保持原始显示方式）
+            self.displayImage(img, self.original_label)
+            self.displayImage(overlay_img, self.result_label)
+            
+            # 生成报告（新增功能）
+            self.saveReport(contours)
+            
+            self.index += 1
+        except Exception as e:
+            print(f"处理错误: {str(e)}")
+
+    def displayImage(self, img, container):
+        # 显示逻辑（与原始代码完全一致）
+        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+        h, w, ch = img.shape
+        bytes_per_line = ch * w
+        q_img = QImage(img.data, w, h, bytes_per_line, QImage.Format_RGB888)
+        pixmap = QPixmap.fromImage(q_img)
+        container.findChild(QLabel).setPixmap(pixmap)
+
+    def saveReport(self, contours):
+        # 报告生成（新增功能）
+        report_data = {
+            "case_id": self.images[self.index][:8],
+            "date": datetime.now().strftime("%Y-%m-%d %H:%M"),
+            "count": len(contours),
+            "sizes": [cv2.contourArea(c)*0.1 for c in contours]
+        }
+        self.report_history.append(report_data)
+        if len(self.report_history) > 20:
+            self.report_history.pop(0)
+
+    def generateReport(self):
+        # 报告显示（新增功能）
+        if self.report_history:
+            latest = self.report_history[-1]
+            report = f"""
+            诊断报告 {latest['date']}
+            ----------------------------
+            病例ID: {latest['case_id']}
+            息肉数量: {latest['count']} 处
+            最大尺寸: {max(latest['sizes']):.1f}mm
+            """
+            self.report_content.setText(report)
+
+    def showHistory(self):
+        # 历史记录查看（新增功能）
+        history_dialog = QDialog(self)
+        history_dialog.setWindowTitle("历史记录")
+        layout = QVBoxLayout()
+        
+        list_widget = QListWidget()
+        for report in self.report_history[-20:]:
+            item = QListWidgetItem(f"{report['case_id']} - {report['date']}")
+            list_widget.addItem(item)
+        
+        layout.addWidget(list_widget)
+        history_dialog.setLayout(layout)
+        history_dialog.exec_()
+
+    def openAIAssistant(self):
+        # AI辅助功能（新增）
+        self.ai_window = QWebEngineView()
+        self.ai_window.load(QUrl("https://www.doubao.com/chat/"))
+        self.ai_window.show()
+
+    def applyStyle(self):
+        # 界面样式（优化部分）
+        self.setStyleSheet("""
+            QWidget {
+                background-color: #F0F8FF;
+                font-family: '微软雅黑';
+            }
+            QGroupBox {
+                font-size: 16px;
+                color: #2C3E50;
+            }
+            QScrollArea {
+                border: 2px solid #BDC3C7;
+                border-radius: 8px;
+            }
+        """)
+
+if __name__ == '__main__':
+    app = QApplication(sys.argv)
+    app.setFont(QFont("Microsoft YaHei", 10))
+    window = MedicalAISystem()
+    window.show()
+    sys.exit(app.exec_())
\ No newline at end of file