Update utils.py

utils.py

@@ -19,6 +19,9 @@ logger = logging.getLogger(__name__)
 # Initialize HF API token (get this from your HF account)
 HF_API_TOKEN = os.environ.get("HF_API_TOKEN", "")
 
+# Initialize HF API token (get this from your HF account)
+HF_API_TOKEN = os.environ.get("HF_API_TOKEN", "")
+
 # Initialize intent classifier
 try:
     intent_classifier = pipeline(
@@ -174,4 +177,38 @@ def text_to_speech(text):
 
 def get_fallback_response():
     """Return a fallback response"""
-    return random.choice(FALLBACK_RESPONSES)
+    return random.choice(FALLBACK_RESPONSES)
+
+def get_rag_response(query, intent, hf_space_url):
+    """Get response using the RAG system via Hugging Face Spaces"""
+    try:
+        # Prepare data for the Hugging Face Space
+        api_url = f"{hf_space_url}/api/predict"
+        payload = {
+            "data": [
+                query,
+                intent
+            ]
+        }
+        
+        # Check if we should use API token
+        headers = {}
+        if HF_API_TOKEN:
+            headers["Authorization"] = f"Bearer {HF_API_TOKEN}"
+        
+        # Call the Hugging Face Space
+        response = requests.post(api_url, json=payload, headers=headers)
+        
+        if response.status_code == 200:
+            result = response.json()
+            # Extract the response text from the result
+            # Structure will depend on your Space's output format
+            response_text = result.get("data", ["I'm sorry, I couldn't process that request."])[0]
+            return response_text
+        else:
+            logger.error(f"Error from HF Space: {response.status_code} - {response.text}")
+            return get_fallback_response()
+            
+    except Exception as e:
+        logger.error(f"Error getting RAG response: {e}")
+        return get_fallback_response()