Update app.py

app.py

@@ -99,6 +99,7 @@ class PredictionResponse(BaseModel):
     generated_text: str
     model_used: str
 
+# Pythhon version check
 @app.get("/version")
 async def version():
     return {
@@ -106,6 +107,26 @@ async def version():
         "models_available": list(MODELS.keys())
     }
 
+# Environment check 
+@app.get("/debug/environment")
+async def get_environment_info():
+    import torch
+    import transformers
+    import platform
+    
+    return {
+        "python_version": platform.python_version(),
+        "torch_version": torch.__version__,
+        "torch_device": str(torch.device("cuda" if torch.cuda.is_available() else "cpu")),
+        "transformers_version": transformers.__version__,
+        "available_devices": {
+            "cuda_available": torch.cuda.is_available(),
+            "cuda_device_count": torch.cuda.device_count() if torch.cuda.is_available() else 0,
+            "mps_available": hasattr(torch.backends, "mps") and torch.backends.mps.is_available()
+        }
+    }
+
+# Health check
 @app.get("/health")
 async def health():
     # More comprehensive health check
@@ -123,6 +144,7 @@ async def health():
             "error": str(e)
         }
 
+# Main /predict endpoint
 @app.post("/predict", response_model=PredictionResponse)
 async def predict(request: PredictionRequest):
     try: