Bitrate problem solved

Author: Özgür Ateş Fırat

ScaleUp/core/VideoConventer.py

@@ -9,16 +9,13 @@ def __init__(self, update_status, src, dest, dest_codec, dest_audio_codec, upsca
         self._input_audio_stream = self._input_container.streams.audio[0] if dest_audio_codec and self._input_container.streams.audio else None
 
         self._upscaler = upscaler
+        self.dest_width, self.dest_height = (
+            self._input_video_stream.width,
+            self._input_video_stream.height
+        )
         if upscaler:
-            self.dest_width, self.dest_height = (
-                self._input_video_stream.width * upscaler.scale,
-                self._input_video_stream.height * upscaler.scale
-            )
-        else:
-            self.dest_width, self.dest_height = (
-                self._input_video_stream.width,
-                self._input_video_stream.height
-            )
+            self.dest_width *= upscaler.scale
+            self.dest_height *= upscaler.scale
 
         self._output_container = av.open(dest, mode="w")
 
@@ -30,6 +27,9 @@ def __init__(self, update_status, src, dest, dest_codec, dest_audio_codec, upsca
         self._output_video_stream.height = self.dest_height
         self._output_video_stream.bit_rate = self._input_video_stream.bit_rate
 
+        if upscaler:
+            self._output_video_stream.bit_rate *= upscaler.scale
+
         self._output_audio_stream = self._output_container.add_stream(
             dest_audio_codec,
             rate=self._input_audio_stream.average_rate

ScaleUp/gui/FileSelection/FileSelection.py

@@ -15,9 +15,12 @@ def __init__(self, parent=None):
 
     def _file_selector(self, event):
         file_dialog = QFileDialog()
-        file_dialog.setFileMode(QFileDialog.ExistingFile if self._existing_file else QFileDialog.AnyFile)
 
-        file_path = file_dialog.getOpenFileName()[0]
+        file_path = ...
+        if self._existing_file:
+            file_path = file_dialog.getOpenFileName()[0]
+        else:
+            file_path = file_dialog.getSaveFileName()[0]
 
         if file_path:
             self.ui.lineEdit.setText(file_path)

ScaleUp/gui/MainWindow.ui

@@ -31,7 +31,11 @@
     </widget>
    </item>
    <item>
-    <widget class="FileSelection" name="sourceFileSelection"/>
+    <widget class="FileSelection" name="sourceFileSelection">
+     <property name="existingFile">
+         <bool>true</bool>
+     </property>
+    </widget>
    </item>
    <item>
     <widget class="QLabel" name="destinationLabel">

setup.py

@@ -37,7 +37,7 @@ def _process_translations(self):
 
 setup(
     name="ScaleUp",
-    version="1.0.1",
+    version="1.0.2",
     description="Real-ESRGAN based video upscaler",
     author="Özgür Ateş Fırat",
     author_email="ozgurafirat@proton.me",