Upload app.py

app.py

@@ -178,25 +178,71 @@ def stream_huggingface(python, model_name):
             timeout=60
         )
         
+        # Check if response body is empty
+        if not response.text or len(response.text.strip()) == 0:
+            yield f"⏳ Model is loading or initializing...\n\n"
+            yield f"This happens on first use. Please try again in 30-60 seconds.\n\n"
+            yield f"💡 Quick alternative: Use GPT-4o or Claude-3.5-Sonnet (instant results!)"
+            return
+        
         if response.status_code == 200:
-            result = response.json()
-            if isinstance(result, list) and len(result) > 0:
-                generated_text = result[0].get("generated_text", "")
-            else:
-                generated_text = result.get("generated_text", "")
-            
-            # Clean up the response
-            reply = generated_text.replace('```cpp\n','').replace('```','')
-            yield reply
+            try:
+                result = response.json()
+                if isinstance(result, list) and len(result) > 0:
+                    generated_text = result[0].get("generated_text", "")
+                else:
+                    generated_text = result.get("generated_text", "")
+                
+                if not generated_text or len(generated_text.strip()) == 0:
+                    yield f"⚠️ Model returned empty response.\n\n"
+                    yield f"Try again or use GPT-4o/Claude-3.5-Sonnet instead."
+                    return
+                
+                # Clean up the response
+                reply = generated_text.replace('```cpp\n','').replace('```','')
+                yield reply
+            except ValueError as json_err:
+                # JSON parsing failed
+                yield f"⚠️ Model response format error.\n\n"
+                yield f"The model might still be warming up. Try again in 30 seconds.\n\n"
+                yield f"💡 Or use GPT-4o/Claude-3.5-Sonnet for instant results!"
+                
+        elif response.status_code == 401 or response.status_code == 403:
+            # Authentication error - need HF token
+            yield f"🔑 Authentication Required!\n\n"
+            yield f"To use open-source models, you need a FREE Hugging Face token:\n\n"
+            yield f"1. Get token: https://huggingface.co/settings/tokens\n"
+            yield f"2. Add HF_TOKEN secret in Space Settings\n"
+            yield f"3. Factory reboot\n\n"
+            yield f"OR use GPT-4o/Claude-3.5-Sonnet instead (they work now!)"
+        elif response.status_code == 503:
+            # Service unavailable - model loading
+            yield f"⏳ Model is currently loading (cold start)...\n\n"
+            yield f"This can take 30-60 seconds on first use.\n"
+            yield f"Please wait a minute and try again.\n\n"
+            yield f"💡 Quick solution: Use GPT-4o or Claude-3.5-Sonnet (no waiting!)"
         else:
-            error_msg = response.json().get("error", "Unknown error")
-            if "loading" in error_msg.lower():
+            try:
+                error_msg = response.json().get("error", "Unknown error")
+            except:
+                error_msg = response.text[:200] if response.text else "Empty response"
+            
+            if "loading" in str(error_msg).lower():
                 yield f"⏳ Model is loading... This may take 20-30 seconds. Please try again."
             else:
-                yield f"❌ Error from Hugging Face: {error_msg}"
+                yield f"❌ Error from Hugging Face (HTTP {response.status_code}):\n{error_msg}\n\n"
+                yield f"💡 Tip: Use GPT-4o or Claude-3.5-Sonnet for now (they're working!)"
             
+    except requests.exceptions.Timeout:
+        yield f"⏱️ Request timed out.\n\n"
+        yield f"Model might be loading (cold start). Try again in 30-60 seconds.\n\n"
+        yield f"💡 Or use GPT-4o/Claude-3.5-Sonnet for instant results!"
+    except requests.exceptions.RequestException as req_err:
+        yield f"🌐 Network error: {str(req_err)}\n\n"
+        yield f"💡 Please use GPT-4o or Claude-3.5-Sonnet instead."
     except Exception as e:
-        yield f"❌ Error calling Hugging Face: {str(e)}"
+        yield f"❌ Unexpected error: {str(e)}\n\n"
+        yield f"💡 Tip: Use GPT-4o or Claude-3.5-Sonnet for reliable results!"
 
 def optimize(python, model):
     """Convert Python to C++ using selected AI model"""