Update app.py

app.py

@@ -294,4 +294,49 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch() 
+    demo.launch() 
+
+    # ... (keep all the previous imports and model loading code)
+
+with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
+    with gr.Column(elem_id="main-container"):
+        # Enhanced Header Section
+        with gr.Column(elem_classes=["header"]):
+            gr.Markdown("""
+            <div style="text-align: center;">
+                <h1 style="font-size: 2.5em; margin-bottom: 0.2em; 
+                    background: linear-gradient(90deg, #4b6cb7 0%, #182848 100%);
+                    -webkit-background-clip: text;
+                    -webkit-text-fill-color: transparent;
+                    display: inline-block;">
+                    🛡️ DeepGuard AI
+                </h1>
+                <p style="font-size: 1.1em; color: #555; margin-top: 0;">
+                    Advanced Deepfake & AI-Generated Content Detection System
+                </p>
+            </div>
+            """)
+        
+        # ... (rest of the interface code remains the same)
+        
+        with gr.Row():
+            with gr.Column(scale=1.5):
+                image_input = gr.Image(
+                    type="pil", 
+                    label="Upload Image for Analysis", 
+                    elem_classes=["upload-area", "animation"]
+                )
+                
+                with gr.Row():
+                    model_selector = gr.Dropdown(
+                        choices=list(MODEL_NAMES.keys()),
+                        value=list(MODEL_NAMES.keys())[0],
+                        label="Select Detection Model",
+                        elem_classes=["model-select", "animation"]
+                    )
+                    analyze_btn = gr.Button(
+                        "Analyze Image", 
+                        elem_classes=["btn-primary", "animation"]
+                    )
+            
+            # ... (rest of the interface code remains the same)