logging

app.py

@@ -17,8 +17,8 @@ from trellis.pipelines import TrellisImageTo3DPipeline
 from trellis.representations import Gaussian, MeshExtractResult
 from trellis.utils import render_utils, postprocessing_utils
 from huggingface_hub import InferenceClient
-from PIL import Image
-import io
+import requests
+import time
 
 MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
@@ -106,6 +106,22 @@ def get_seed(randomize_seed: bool, seed: int) -> int:
     """
     return np.random.randint(0, MAX_SEED) if randomize_seed else seed
 
+def test_connection():
+    """Test basic internet connectivity"""
+    try:
+        # Test general internet connectivity
+        response = requests.get("https://www.google.com", timeout=5)
+        logger.info(f"Google connection test: {response.status_code}")
+        
+        # Test HF connection
+        response = requests.get("https://huggingface.co", timeout=5)
+        logger.info(f"HuggingFace connection test: {response.status_code}")
+
+        return True
+    except Exception as e:
+        logger.error(f"Connection test failed: {str(e)}")
+        return False
+
 
 def text_to_image(
     prompt: str,
@@ -124,24 +140,80 @@ def text_to_image(
         if not api_token:
             logger.error("API token not found")
             raise gr.Error("API token not found. Please set HF_API_TOKEN environment variable.")
+        
+        # Test connection before making the main request
+        if not test_connection():
+            raise gr.Error("Network connectivity issues detected")
             
-        client = InferenceClient(
-            model="black-forest-labs/FLUX.1-schnell",
-            token=api_token
-        )
+        headers = {
+            "Authorization": f"Bearer {api_token}",
+            "Content-Type": "application/json"
+        }
         
+        payload = {
+            "inputs": prompt,
+            "parameters": {
+                "height": height,
+                "width": width,
+                "guidance_scale": guidance_scale,
+                "num_inference_steps": num_inference_steps,
+            }
+        }
+
         logger.info("Sending request to HF Inference API...")
-        image = client.text_to_image(
-            prompt,
-            height=height,
-            width=width,
-            guidance_scale=guidance_scale,
-            num_inference_steps=num_inference_steps,
-        )
         
+        # Use session for better connection handling
+        with requests.Session() as session:
+            # First test the API endpoint
+            try:
+                test_response = session.get(
+                    "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell",
+                    timeout=5
+                )
+                logger.info(f"API status test: {test_response.status_code}")
+            except Exception as e:
+                logger.error(f"API status test failed: {str(e)}")
+            
+            # Main request with retries
+            for attempt in range(3):
+                try:
+                    response = session.post(
+                        "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell",
+                        headers=headers,
+                        json=payload,
+                        timeout=(10, 30)  # (connect timeout, read timeout)
+                    )
+                    break
+                except requests.exceptions.Timeout:
+                    if attempt == 2:  # Last attempt
+                        raise
+                    logger.warning(f"Attempt {attempt + 1} timed out, retrying...")
+                    time.sleep(2)  # Wait before retry
+        
+        logger.info(f"Response status code: {response.status_code}")
+        
+        if response.status_code == 503:
+            logger.warning("Model is loading...")
+            raise gr.Error("The model is currently loading. Please try again in a few moments.")
+        elif response.status_code != 200:
+            logger.error(f"Error response: {response.text}")
+            raise gr.Error(f"API request failed with status {response.status_code}")
+        
+        # Log response content type and size
+        logger.info(f"Response content type: {response.headers.get('content-type')}")
+        logger.info(f"Response size: {len(response.content)} bytes")
+        
+        # Try to open the image and log success
+        image = Image.open(io.BytesIO(response.content))
         logger.info(f"Successfully created image: {image.size} {image.mode}")
         return image
         
+    except requests.exceptions.Timeout:
+        logger.error("Request timed out")
+        raise gr.Error("Request timed out. The server took too long to respond.")
+    except requests.exceptions.ConnectionError as e:
+        logger.error(f"Connection error: {str(e)}")
+        raise gr.Error("Failed to connect to the server. Please check your network connection.")
     except Exception as e:
         logger.exception("Error in text_to_image")
         raise gr.Error(f"Failed to generate image: {str(e)}")

requirements.txt

@@ -21,7 +21,6 @@ xformers==0.0.27.post2
 spconv-cu120==2.3.6
 transformers==4.46.3
 gradio_litmodel3d==0.0.1
-huggingface-hub>=0.20.3
 https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
 https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true