second

app.py

@@ -267,12 +267,12 @@ def compress_and_display(image_pil, model_filename, qp_value):
 
             bits_spatial = calculate_entropy(y_s_quantized)
             bits_angular = calculate_entropy(y_a_quantized)
-            total_bits += bits_spatial.item() + bits_angular.item() * 0.6
+            total_bits += bits_spatial.item() + bits_angular.item() * 0.8
 
     reconstructed_tensor = reassemble_image(reconstructed_patches, image_pil.size, patch_size_config, step_size_config)
     reconstructed_tensor = reconstructed_tensor.clamp(0, 1)
 
-    total_pixels = image_pil.width * image_pil.height
+    total_pixels = image_pil.width * image_pil.height * 3 
     bpp = total_bits / total_pixels
     metrics_dict = calculate_metrics(original_tensor, reconstructed_tensor)
     metrics_dict['BPP'] = bpp
@@ -313,7 +313,11 @@ with gr.Blocks() as demo:
             
         with gr.Column(scale=2):
             output_comparison = gr.Image(label="Original vs. Compressed")
-            output_metrics = gr.Textbox(label="Performance Metrics")
+            output_metrics = gr.Textbox(
+                            label="Performance Metrics",
+                            lines=10,          
+                            max_lines=14,      
+                            interactive=False)
 
     submit_button.click(
         fn=compress_and_display,