Update app.py

app.py

@@ -9,7 +9,7 @@ __import__('pysqlite3')
 sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
 
 from crewai import Agent, Task, Crew
-from langchain_huggingface import HuggingFaceEndpoint
+from langchain_community.llms import HuggingFaceHub
 
 def generate_jingle(theme):
     # Hugging Face API Key handling
@@ -19,20 +19,21 @@ def generate_jingle(theme):
     if not hf_api_key.startswith("hf_"):
         return "Error: Invalid Hugging Face API token. It should start with 'hf_'. Please check and update the token in Space secrets."
 
-    # LLM setup with Mistral 7B
+    # LLM setup with HuggingFaceHub
     try:
-        llm = HuggingFaceEndpoint(
-            endpoint_url="https://api-inference.huggingface.co/models/mixtral-8x7b-instruct-v0.1",
+        llm = HuggingFaceHub(
+            repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
             huggingfacehub_api_token=hf_api_key,
-            max_new_tokens=500,
-            temperature=0.7,
-            top_p=0.9,
-            retry_on_rate_limit=True,
-            timeout=30
+            model_kwargs={
+                "max_new_tokens": 500,
+                "temperature": 0.7,
+                "top_p": 0.9
+            }
         )
         # Test the endpoint
         try:
             test_response = llm.invoke("Test connection to Hugging Face API")
+            return f"Hugging Face API connection successful! Response: {test_response[:100]}...\n\nStarting jingle generation..."
         except Exception as test_e:
             return f"Warning: Test API call failed: {str(test_e)}. Proceeding anyway, but jingle generation may fail."
     except Exception as e: