V0

app.py

@@ -615,7 +615,8 @@ def create_gradio_interface():
                     with gr.Accordion("Analysis Details", open=True):
                         output_text = gr.Markdown(label="Detection Results")
         
-        # Diagnostics Tab
+ # This is the continuation of the code, starting from the System Diagnostics tab
+
         with gr.Tab("System Diagnostics"):
             with gr.Row():
                 run_diagnostic_btn = gr.Button("Run Diagnostics", variant="primary")
@@ -635,6 +636,21 @@ 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):
@@ -704,8 +720,8 @@ def create_gradio_interface():
                 output.append("\n**Note:** You may need to restart the application for changes to take effect.")
                 
                 return "\n".join(output)
-                
-               # Settings Tab
+        
+        # Settings Tab
         with gr.Tab("Settings"):
             gr.Markdown("## Model Settings")
             
@@ -770,7 +786,7 @@ def create_gradio_interface():
                 gr.Examples(
                     examples=example_inputs,
                     inputs=input_image,
-                    outputs=[output_image, output_text],
+                    outputs=[output_image, output_text, usage_counter],  # Add usage_counter to outputs
                     fn=lambda x: process_image(
                         x, DEFAULT_DAMAGE_MODEL_PATH, DEFAULT_DEEPFAKE_MODEL_PATH, 
                         0.7, 0.5, False, "auto", 0  # Added initial usage_count value of 0
@@ -831,7 +847,9 @@ def create_gradio_interface():
     return app
 
 if __name__ == "__main__":
+    # Check if dependencies are installed
+    auto_install_dependencies()
     
     # Create and launch the Gradio app
     app = create_gradio_interface()
-    app.launch(share=False)  # Set share=True to create a public linkf"**PyTorch:** {torch.__version__}")
+    app.launch(share=False)  # Set share=True to create a public link