V0

app.py

@@ -635,21 +635,6 @@ def create_gradio_interface():
                 
                 try:
                     import torch
-                    output.append(f"**PyTorch:** {torch.__version__}")
-                    output.append(f"**CUDA Available:** {'Yes ✅' if torch.cuda.is_available() else 'No ❌'}")
-                    if torch.cuda.is_available():
-                        output.append(f"**GPU:** {torch.cuda.get_device_name(0)}")
-                    output.append(f"**MPS Available:** {'Yes ✅' if hasattr(torch.backends, 'mps') and torch.backends.mps.is_available() else 'No ❌'}")
-                except ImportError:
-                    output.append("**PyTorch:** Not installed ❌")
-                
-                output.append("\n## Model Files")
-                
-                # Check model files
-                if os.path.exists(DEFAULT_DAMAGE_MODEL_PATH):
-                    file_size = os.path.getsize(DEFAULT_DAMAGE_MODEL_PATH) / (1024 * 1024)  # Size in MB
-                    output.append(f"**Damage Model:** Available ✅ ({file_size:.2f} MB)")
-                else:
                     output.append(f"**Damage Model:** Not found ❌ at {DEFAULT_DAMAGE_MODEL_PATH}")
                 
                 if os.path.exists(DEFAULT_DEEPFAKE_MODEL_PATH):
@@ -788,7 +773,7 @@ def create_gradio_interface():
                     outputs=[output_image, output_text],
                     fn=lambda x: process_image(
                         x, DEFAULT_DAMAGE_MODEL_PATH, DEFAULT_DEEPFAKE_MODEL_PATH, 
-                        0.7, 0.5, False, "auto"
+                        0.7, 0.5, False, "auto", 0  # Added initial usage_count value of 0
                     ),
                     cache_examples=True
                 )
@@ -803,16 +788,17 @@ def create_gradio_interface():
                 damage_threshold,
                 deepfake_threshold,
                 skip_damage,
-                device
+                device,
+                usage_counter  # Add the usage counter state here
             ],
-            outputs=[output_image, output_text]
+            outputs=[output_image, output_text, usage_counter]  # Update usage counter in outputs
         )
         
         # Clear button functionality
         clear_btn.click(
-            fn=lambda: [None, ""],
+            fn=lambda: [None, "", 0],  # Reset usage counter to 0 when clearing
             inputs=[],
-            outputs=[output_image, output_text]
+            outputs=[output_image, output_text, usage_counter]
         )
         
         # System diagnostic buttons
@@ -835,10 +821,17 @@ def create_gradio_interface():
             outputs=paths_result
         )
         
+        # Update usage display when counter changes
+        usage_counter.change(
+            fn=lambda count: f"**Usage: {count}/{MAX_TRIES}**",
+            inputs=[usage_counter],
+            outputs=[usage_display]
+        )
+        
     return app
 
 if __name__ == "__main__":
     
     # Create and launch the Gradio app
     app = create_gradio_interface()
-    app.launch(share=False)  # Set share=True to create a public link#!/usr/bin/env python3
+    app.launch(share=False)  # Set share=True to create a public linkf"**PyTorch:** {torch.__version__}")