Update app.py

app.py

@@ -1,36 +1,27 @@
 import os
 import gradio as gr
 import requests
-from PIL import Image
-from io import BytesIO
 import logging
 
-# Configure logging
+# Setup logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Get credentials from environment variables
-RUNPOD_API_KEY = os.getenv("RUNPOD_API_KEY")
-RUNPOD_ENDPOINT = os.getenv("RUNPOD_ENDPOINT")
-
-def generate_single_image(prompt):
-    """Barebones image generation with maximum error handling"""
-    if not RUNPOD_API_KEY or not RUNPOD_ENDPOINT:
-        error = "API credentials not set! Check your Space secrets."
-        logger.error(error)
-        return None, error
-    
+def test_connection():
+    """Simplest possible connectivity test"""
     try:
-        # Basic payload - adjust based on your RunPod template
+        RUNPOD_API_KEY = os.getenv("RUNPOD_API_KEY")
+        RUNPOD_ENDPOINT = os.getenv("RUNPOD_ENDPOINT")
+        
+        if not RUNPOD_API_KEY or not RUNPOD_ENDPOINT:
+            return "❌ Missing API key or endpoint in secrets"
+        
+        test_prompt = "A simple test image"
         payload = {
             "input": {
-                "prompt": f"Comic book style: {prompt}",
-                "negative_prompt": "blurry, deformed, text, watermark",
-                "width": 768,
-                "height": 768,
-                "num_inference_steps": 25,
-                "guidance_scale": 7.5,
-                "return_base64": True  # More reliable than URLs in Spaces
+                "prompt": test_prompt,
+                "width": 64,
+                "height": 64  # Minimal size for quick test
             }
         }
         
@@ -39,56 +30,31 @@ def generate_single_image(prompt):
             "Content-Type": "application/json"
         }
         
-        logger.info(f"Sending request to: {RUNPOD_ENDPOINT}")
+        logger.info(f"Testing endpoint: {RUNPOD_ENDPOINT}")
         response = requests.post(
             RUNPOD_ENDPOINT,
             json=payload,
             headers=headers,
-            timeout=120
+            timeout=10
         )
         
-        logger.info(f"Received status: {response.status_code}")
-        
-        if response.status_code != 200:
-            error = f"API Error {response.status_code}: {response.text}"
-            logger.error(error)
-            return None, error
-            
-        data = response.json()
-        logger.info("Got successful API response")
-        
-        # Handle both base64 and URL responses
-        if "image" in data:
-            image_data = base64.b64decode(data["image"].split(",")[1])
-            return Image.open(BytesIO(image_data)), "Success!"
-        elif "output" in data and isinstance(data["output"], list):
-            image_url = data["output"][0]
-            img_data = requests.get(image_url, timeout=60).content
-            return Image.open(BytesIO(img_data)), "Success!"
+        if response.status_code == 200:
+            return "✅ Connection successful! Response: " + str(response.json())[:100] + "..."
         else:
-            error = f"Unexpected response format: {data}"
-            logger.error(error)
-            return None, error
+            return f"❌ API Error {response.status_code}: {response.text}"
             
     except Exception as e:
-        error = f"Generation failed: {str(e)}"
-        logger.error(error, exc_info=True)
-        return None, error
+        return f"🔥 Critical failure: {str(e)}"
 
-# Simple interface
 with gr.Blocks() as demo:
-    gr.Markdown("# 🚀 RunPod Tester")
-    with gr.Row():
-        prompt = gr.Textbox(label="Prompt", value="A superhero standing on a rooftop")
-        btn = gr.Button("Generate")
-    output = gr.Image(label="Result")
-    status = gr.Textbox(label="Status")
-
-    btn.click(
-        generate_single_image,
-        inputs=[prompt],
-        outputs=[output, status]
+    gr.Markdown("# 🚨 RunPod Connection Tester")
+    test_btn = gr.Button("Test Connection")
+    output = gr.Textbox(label="Diagnostic Output")
+    
+    test_btn.click(
+        test_connection,
+        inputs=[],
+        outputs=[output]
     )
 
-if __name__ == "__main__":
-    demo.launch(debug=True)
+demo.launch(debug=True)