Update app.py

app.py

@@ -46,6 +46,21 @@ def get_client():
 CLIENT, STATUS = get_client()
 logger.info(f"Client status: {STATUS}")
 
+# Add debug info
+if CLIENT:
+    logger.info("✅ Client successfully initialized with provider='hf-inference'")
+    
+    # Test the client with a simple call
+    try:
+        test_result = CLIENT.fill_mask("The capital of France is [MASK].", model="google-bert/bert-base-uncased")
+        logger.info(f"✅ Client test successful: {type(test_result)}")
+    except Exception as e:
+        logger.error(f"❌ Client test failed: {e}")
+        STATUS = f"Client test failed: {e}"
+        CLIENT = None
+else:
+    logger.error(f"❌ Client initialization failed: {STATUS}")
+
 # ══════════════════════════════════════════════════════════════════════════════
 # 🤖 SIMPLE TASK FUNCTIONS (No Async)
 # ══════════════════════════════════════════════════════════════════════════════
@@ -82,7 +97,7 @@ def speech_recognition(audio_file):
         return f"❌ Error: {str(e)}"
 
 def chat_completion(messages_json):
-    """Chat Completion - Following official docs exactly."""
+    """Chat Completion - Using exact working model from your test."""
     if not messages_json.strip():
         return "❌ Please enter valid JSON messages"
     
@@ -92,9 +107,9 @@ def chat_completion(messages_json):
     try:
         messages = json.loads(messages_json)
         
-        # Official API call from docs
+        # Using the EXACT same model and call that works in your test
         completion = CLIENT.chat.completions.create(
-            model="meta-llama/Llama-3.2-3B-Instruct",
+            model="Qwen/Qwen2.5-72B-Instruct",  # Using available Qwen model
             messages=messages,
         )
         
@@ -103,10 +118,10 @@ def chat_completion(messages_json):
     except json.JSONDecodeError:
         return "❌ Invalid JSON format"
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        return f"❌ Error: {str(e)}\n\n**Debug:** Model: Qwen/Qwen2.5-72B-Instruct, Messages: {messages}"
 
 def fill_mask(text):
-    """Fill Mask - Following official docs exactly."""
+    """Fill Mask - Following official docs exactly with better debugging."""
     if not text or "[MASK]" not in text:
         return "❌ Text must contain [MASK] token"
     
@@ -114,12 +129,16 @@ def fill_mask(text):
         return f"❌ Client not available: {STATUS}"
     
     try:
+        logger.info(f"Making fill_mask call with text: {text}")
+        
         # Official API call from docs
         result = CLIENT.fill_mask(
             text,
             model="google-bert/bert-base-uncased",
         )
         
+        logger.info(f"fill_mask result type: {type(result)}, content: {result}")
+        
         if isinstance(result, list):
             predictions = []
             for i, pred in enumerate(result[:5], 1):
@@ -131,7 +150,8 @@ def fill_mask(text):
             return f"🎭 **Result:** {result}"
             
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        logger.error(f"fill_mask error: {e}")
+        return f"❌ Error: {str(e)}\n\n**Debug:** Model: google-bert/bert-base-uncased, Text: {text}"
 
 def question_answering(question, context):
     """Question Answering - Following official docs exactly."""
@@ -180,7 +200,7 @@ def summarization(text):
         return f"❌ Error: {str(e)}"
 
 def text_generation(prompt):
-    """Text Generation - Following official docs exactly."""
+    """Text Generation - Using working Qwen model."""
     if not prompt.strip():
         return "❌ Prompt cannot be empty"
     
@@ -188,16 +208,16 @@ def text_generation(prompt):
         return f"❌ Client not available: {STATUS}"
     
     try:
-        # Official API call from docs
+        # Using the same model family that works in your test
         completion = CLIENT.chat.completions.create(
-            model="meta-llama/Llama-3.2-3B-Instruct",
+            model="Qwen/Qwen2.5-72B-Instruct",
             messages=[{"role": "user", "content": prompt}],
         )
         
         return f"✍️ **Generated:** {completion.choices[0].message.content}"
         
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        return f"❌ Error: {str(e)}\n\n**Debug:** Model: Qwen/Qwen2.5-72B-Instruct, Prompt: {prompt}"
 
 def image_classification(image_path):
     """Image Classification - Following official docs exactly."""