Retry code added for better responses

app.py

@@ -1,10 +1,12 @@
 """gradio app for Chat Interface for DataStax Langflow calls"""
 
+from requests.exceptions import ConnectTimeout, ReadTimeout
 import re
 from datetime import datetime
 from typing import Optional, Sequence, Tuple
 from uuid import uuid4
 import os
+from retry import retry
 import requests
 import gradio as gr
 from dotenv import load_dotenv
@@ -86,6 +88,7 @@ def submit_signup(_email: str, _name: str, _contact: str):
     return 'Success'
 
 
+@retry((ConnectTimeout, ReadTimeout), tries=3, delay=2)
 def call_travel_ai(
     message: str,
     history: Sequence[Tuple[str, str]],
@@ -125,7 +128,7 @@ def call_travel_ai(
         api_url,
         json=payload,
         headers=headers,
-        timeout=360
+        timeout=120,
     )
     response = response.json()
     resp_message = ''

requirements.txt

@@ -4,3 +4,4 @@ requests
 numpy
 cassandra-driver
 python-dotenv
+retry