Upload app.py with huggingface_hub

app.py

@@ -114,18 +114,32 @@ def call_deepseek_api(messages: List[Dict], client_info: Dict, max_retries: int
     try:
         client = client_info["client"]
         
-        # Use InferenceClient.chat_completion method
-        response = client.chat_completion(
-            messages=messages,
-            max_tokens=1024,
+        # Convert messages to a single prompt for text_generation
+        conversation = ""
+        for msg in messages:
+            if msg["role"] == "system":
+                conversation += f"System: {msg['content']}\n\n"
+            elif msg["role"] == "user":
+                conversation += f"User: {msg['content']}\n\n"
+            elif msg["role"] == "assistant":
+                conversation += f"Assistant: {msg['content']}\n\n"
+        
+        conversation += "Assistant: "
+        
+        # Use text_generation method instead of chat_completion
+        response = client.text_generation(
+            prompt=conversation,
+            max_new_tokens=1024,
             temperature=0.7,
             top_p=0.9,
+            do_sample=True,
+            return_full_text=False,
             stream=False
         )
         
         # Extract content from response
-        if hasattr(response, 'choices') and len(response.choices) > 0:
-            content = response.choices[0].message.content
+        if isinstance(response, str):
+            content = response
             logger.info(f"✅ Success with InferenceClient: {client_info['name']} ({client_info['model']})")
             return content.strip()
         else: