connection attempt update

app.py

@@ -1,3 +1,41 @@
+
+# # asynchronous function
+# async def summarize(text, retries=3):
+#     if len(text)<250:
+#         return "⛔ You input is too short! Please input a longer text for summarization."
+#     headers = {"Authorization": f"Bearer {SECRET_KEY}"}
+#     data = {
+#         "inputs": text,
+#         "options": {
+#             "max_new_tokens": 100,
+#             "min_length": 10,
+#             "max_length": 500,
+#             "length_penalty": 2.0,
+#             "num_beams": 3,
+#             "early_stopping": True
+#         }
+#     }
+#     async with aiohttp.ClientSession() as session:
+#         for attempt in range(retries):
+#             try:
+#                 async with session.post(API_URL, headers=headers, json=data) as response:
+#                     # raise exception for non-200 status codes
+#                     response.raise_for_status()  
+#                     result = await response.json()
+#                     return result[0]["summary_text"]
+#             except aiohttp.ClientResponseError as e:
+#                 # retry only if the error is 503 (Service Unavailable)
+#                 if e.status == 503 and attempt < retries - 1:
+#                     # print(f"Retry attempt {attempt+1} for 503 error.")
+#                     # add a short delay before retrying
+#                     await asyncio.sleep(1)  
+#                     continue
+#                 else:
+#                     return "Oops! 🙈 It looks like those mischievous monkeys🐒 might be swinging around the server, causing a bit of chaos with the cables. Please try it again!"
+#                     # return f"Error: {e.status}, message='{e.message}'. Please try again 🔁"
+
+
+
 import gradio as gr
 import aiohttp
 import asyncio
@@ -9,7 +47,7 @@ SECRET_KEY = os.environ.get("summarizer")
 
 # asynchronous function
 async def summarize(text, retries=3):
-    if len(text)<250:
+    if len(text) < 250:
         return "⛔ You input is too short! Please input a longer text for summarization."
     headers = {"Authorization": f"Bearer {SECRET_KEY}"}
     data = {
@@ -28,19 +66,19 @@ async def summarize(text, retries=3):
             try:
                 async with session.post(API_URL, headers=headers, json=data) as response:
                     # raise exception for non-200 status codes
-                    response.raise_for_status()  
+                    response.raise_for_status()
                     result = await response.json()
                     return result[0]["summary_text"]
-            except aiohttp.ClientResponseError as e:
+            except (aiohttp.ClientResponseError, aiohttp.ClientConnectionError) as e:
                 # retry only if the error is 503 (Service Unavailable)
-                if e.status == 503 and attempt < retries - 1:
-                    # print(f"Retry attempt {attempt+1} for 503 error.")
+                if isinstance(e, aiohttp.ClientResponseError) and e.status == 503 and attempt < retries - 1:
                     # add a short delay before retrying
-                    await asyncio.sleep(1)  
+                    await asyncio.sleep(1)
                     continue
                 else:
                     return "Oops! 🙈 It looks like those mischievous monkeys🐒 might be swinging around the server, causing a bit of chaos with the cables. Please try it again!"
-                    # return f"Error: {e.status}, message='{e.message}'. Please try again 🔁"
+
+
 
 # define gradio interface
 iface = gr.Interface(