fix: back to progress dialog

Author: anisiosts

revengai/utils/periodic_check.py

@@ -1,4 +1,5 @@
 from threading import Timer
+import threading
 from os.path import basename
 from typing import Optional
 from binaryninja import log_info, log_error, BinaryView
@@ -105,7 +106,7 @@ def start_ai_decompiler_checking(self, function_id: int, callback, editor: QPlai
             self._ai_decompiler_timer.timeout.connect(lambda: self._ai_decompiler_worker(function_id, name, callback, editor))
 
             # Start the timer with 5 second intervals
-            self._ai_decompiler_timer.start(5000)  # 1000 ms = 1 second
+            self._ai_decompiler_timer.start(30000)  # 1000 ms = 1 second
 
             log_info(f"RevEng.AI | Started AI decompiler periodic check for: {name}")
 

revengai/utils/progress_dialog.py

@@ -123,7 +123,90 @@ def worker_thread():
 
                 if not res.get("status", False):
                     log_info(f"RevEng.AI | AI Decompilation for function at 0x{function_id:x} failed")
-                    self.update_text_signal.emit(self._current_callback, f"AI Decompilation failed: {res.get('errors').get('message')}")
+                    self.update_text_signal.emit(self._current_callback, from PySide6.QtWidgets import QProgressDialog, QProgressBar, QPushButton
+from PySide6.QtCore import Qt
+
+def create_progress_dialog(parent, title, message):
+    progress = QProgressDialog(message, None, 0, 0, parent)
+    progress.setWindowTitle(title)
+    progress.setWindowModality(Qt.WindowModal)
+    progress.setCancelButton(None)
+    progress.setMinimumWidth(400)
+    progress.setMinimumHeight(100)
+    
+    progress_bar = progress.findChild(QProgressBar)
+    if progress_bar:
+        progress_bar.setMinimumWidth(250)
+        progress_bar.setMinimumHeight(20)
+    
+    progress.setStyleSheet("""
+        QProgressBar {
+            border: 1px solid #cccccc;
+            border-radius: 4px;
+            text-align: center;
+            background-color: #f0f0f0;
+            min-width: 250px;
+            min-height: 20px;
+        }
+        QProgressBar::chunk {
+            background-color: #007bff;
+            border-radius: 3px;
+        }
+    """)
+    
+    return progress
+
+def create_cancellable_progress_dialog(parent, title, message, cancel_callback=None):
+    """Create a progress dialog with a cancel button that can stop threads"""
+    progress = QProgressDialog(message, "Cancel", 0, 0, parent)
+    progress.setWindowTitle(title)
+    progress.setWindowModality(Qt.WindowModal)
+    progress.setMinimumWidth(400)
+    progress.setMinimumHeight(100)
+    
+    # Style the cancel button
+    cancel_button = progress.findChild(QPushButton)
+    if cancel_button:
+        cancel_button.setStyleSheet("""
+            QPushButton {
+                background-color: #dc3545;
+                color: white;
+                padding: 6px 12px;
+                border-radius: 4px;
+                border: none;
+                min-width: 60px;
+            }
+            QPushButton:hover {
+                background-color: #c82333;
+            }
+            QPushButton:pressed {
+                background-color: #bd2130;
+            }
+        """)
+
+    cancel_button.clicked.connect(cancel_callback)
+    
+    progress_bar = progress.findChild(QProgressBar)
+    if progress_bar:
+        progress_bar.setMinimumWidth(250)
+        progress_bar.setMinimumHeight(20)
+    
+    progress.setStyleSheet("""
+        QProgressBar {
+            border: 1px solid #cccccc;
+            border-radius: 4px;
+            text-align: center;
+            background-color: #f0f0f0;
+            min-width: 250px;
+            min-height: 20px;
+        }
+        QProgressBar::chunk {
+            background-color: #007bff;
+            border-radius: 3px;
+        }
+    """)
+
+    return progress f"AI Decompilation failed: {res.get('errors').get('message')}")
                     self.stop()
                     return