Improved saveImage feature

1 files changed, 6 insertions, 2 deletions

diff --git a/src/main/python/main.py b/src/main/python/main.py
index 23ec01f..de8a12b 100644
--- a/src/main/python/main.py
+++ b/src/main/python/main.py
@@ -26,7 +26,10 @@ class appWindow(QMainWindow):
         
         #create the menu bar
         self.createMenuBar()
-        
+
+        #used for file name
+        self.counterr = 0
+
         self.mdi = QMdiArea(self) #create area for files to be displayed
         self.mdi.setObjectName('mdi area')
         
@@ -143,13 +146,14 @@ class appWindow(QMainWindow):
         if self.mdi.currentSubWindow():
             currentDiagram = self.mdi.currentSubWindow().tabber.currentWidget().painter
             if currentDiagram:
-                name = QFileDialog.getSaveFileName(self, 'Save File', 'New Image', 'PNG (*.png);;JPEG (*.jpg)')
+                name = QFileDialog.getSaveFileName(self, 'Save File', f'New Image {self.counterr}', 'PNG (*.png);;JPEG (*.jpg)')
                 if name[0]:
                     image = QImage(currentDiagram.sceneRect().size().toSize(), QImage.Format_ARGB32)
                     image.fill(Qt.transparent)
                     painter = QPainter(image)
                     currentDiagram.render(painter)
                     image.save(name[0])
+                    self.counterr += 1
                     painter.end()
     
     def generateReport(self):