Update app.py

app.py

@@ -3,7 +3,7 @@ print("Starting AI Headshot Generator...")
 
 import gradio as gr
 import numpy as np
-from huggingface_hub import spaces  # <-- THIS IS REQUIRED FOR ZeroGPU
+import spaces  # ZeroGPU decorator comes from this package, not huggingface_hub
 
 # ======================
 # Simple license system
@@ -11,6 +11,7 @@ from huggingface_hub import spaces  # <-- THIS IS REQUIRED FOR ZeroGPU
 ADMIN_KEYS = {"HEADSHOT-TEST123456", "HEADSHOT-OWNERACCESS"}
 MAX_TRIALS = 3
 
+
 def verify_license(key):
     if not key:
         return False
@@ -18,12 +19,12 @@ def verify_license(key):
 
 
 # ======================
-# THIS IS THE MAGIC LINE → fixes the ZeroGPU error
+# ZeroGPU-decorated function
 # ======================
-@spaces.GPU(duration=180)   # 180 seconds max per generation (adjust as needed)
+@spaces.GPU(duration=180)  # 180 seconds max per generation
 def generate_headshot(image, license_key):
     """
-    This function will now receive a real GPU when running on ZeroGPU Spaces.
+    This function runs on a real GPU when executed in a ZeroGPU Space.
     Replace the placeholder logic later with your real InstantID / IP-Adapter code.
     """
     # ------------------ License check ------------------
@@ -32,15 +33,15 @@ def generate_headshot(image, license_key):
         return blank, "Invalid license. Use: HEADSHOT-TEST123456"
 
     # ------------------ Real inference goes here ------------------
-    # Example placeholder — replace with your actual pipeline later:
     if image is None:
         h, w = 512, 512
     else:
         h, w = image.shape[:2]
 
-    # Light blue placeholder so you instantly see it works
+    # Light blue/gray placeholder so you instantly see it works
     result = np.ones((h, w, 3), dtype=np.uint8) * 200
-    result[50:150, 50:450] = [255, 182, 193]  # pink rectangle just for fun
+    # Pink rectangle just for fun
+    result[50:150, 50:450] = [255, 182, 193]
 
     return result, "Generated! GPU is working — ready for InstantID pipeline!"
 
@@ -49,10 +50,12 @@ def generate_headshot(image, license_key):
 # Gradio Interface
 # ======================
 with gr.Blocks(title="AI Headshot Generator", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("""
-    # AI Headshot Generator
-    *Transform selfies into professional headshots*
-    """)
+    gr.Markdown(
+        """
+        # AI Headshot Generator
+        *Transform selfies into professional headshots*
+        """
+    )
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -63,17 +66,19 @@ with gr.Blocks(title="AI Headshot Generator", theme=gr.themes.Soft()) as demo:
             license_input = gr.Textbox(
                 label="",
                 value="HEADSHOT-TEST123456",
-                placeholder="Enter HEADSHOT-TEST123456"
+                placeholder="Enter HEADSHOT-TEST123456",
             )
 
-            gr.Markdown(f"""
-            ### How It Works
-            1. Upload a photo  
-            2. Use test key: **HEADSHOT-TEST123456**  
-            3. Get {MAX_TRIALS} free generations  
-            4. Purchase license → HD + multiple styles  
-            [Purchase License](https://canadianheadshotpro.carrd.co)
-            """)
+            gr.Markdown(
+                f"""
+                ### How It Works
+                1. Upload a photo  
+                2. Use test key: **HEADSHOT-TEST123456**  
+                3. Get {MAX_TRIALS} free generations  
+                4. Purchase license → HD + multiple styles  
+                [Purchase License](https://canadianheadshotpro.carrd.co)
+                """
+            )
 
             generate_btn = gr.Button("Generate Headshot", variant="primary", size="lg")
             status_output = gr.Textbox(label="Status", interactive=False)
@@ -82,28 +87,34 @@ with gr.Blocks(title="AI Headshot Generator", theme=gr.themes.Soft()) as demo:
             gr.Markdown("### Your Headshot")
             result_image = gr.Image(label="", height=500)
 
-            gr.Markdown("""
-            ### Download Options
-            - **Free Trial**: Basic preview
-            - **Premium**: HD quality
-            - **Professional**: Multiple styles
-            AI models loading soon!
-            """)
+            gr.Markdown(
+                """
+                ### Download Options
+                - **Free Trial**: Basic preview  
+                - **Premium**: HD quality  
+                - **Professional**: Multiple styles  
+
+                AI models loading soon!
+                """
+            )
 
     # Connect button
     generate_btn.click(
         fn=generate_headshot,
         inputs=[image_input, license_input],
-        outputs=[result_image, status_output]
+        outputs=[result_image, status_output],
     )
 
     # Footer
-    gr.Markdown("""
-    ---
-    <div style="text-align: center; color: #888; margin-top: 20px;">
+    gr.Markdown(
+        """
+        ---
+        <div style="text-align: center; color: #888; margin-top: 20px;">
         Powered by Hugging Face ZeroGPU • InstantID coming soon • © 2025 Bipin Pandey
-    </div>
-    """)
+        </div>
+        """
+    )
+
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()