Update pages/summarizer.py

pages/summarizer.py

@@ -1,6 +1,8 @@
 import streamlit as st
 import os
 import requests
+import time
+
 
 # Define Hugging Face API details
 API_URL = "https://api-inference.huggingface.co/models/Huzaifa367/chat-summarizer"
@@ -9,13 +11,27 @@ HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}
 
 # Function to query Hugging Face API
 def query_huggingface(payload):
-    try:
-        response = requests.post(API_URL, headers=HEADERS, json=payload)
-        response.raise_for_status()  # Raise exception for non-2xx status codes
-        return response.json()
-    except requests.exceptions.RequestException as e:
-        st.error(f"Error querying Hugging Face API: {e}")
-        return {"summary_text": f"Error querying Hugging Face API: {e}"}
+     max_retries=3
+    retries = 0
+    while retries < max_retries:
+        try:
+            response = requests.post(API_URL, headers=HEADERS, json=payload)
+            response.raise_for_status()  # Raise exception for non-2xx status codes
+            return response.json()
+        except requests.exceptions.RequestException as e:
+            st.error(f"Error querying Hugging Face API: {e}")
+            return {"summary_text": f"Error querying Hugging Face API: {e}"}
+        except requests.exceptions.HTTPError as e:
+            if response.status_code == 503:
+                st.warning("Service temporarily unavailable. Retrying...")
+                time.sleep(5)  # Wait for a few seconds before retrying
+                retries += 1
+            else:
+                st.error(f"HTTP Error querying Hugging Face API: {e.response.status_code}")
+                return {"summary_text": "Service Unavailable. Please try again later."}
+        except Exception as e:
+            st.error(f"An unexpected error occurred: {e}")
+            return {"summary_text": "An unexpected error occurred. Please try again later."}
 
 def main():
     st.set_page_config(layout="wide")