Adding status

app.py

@@ -4,6 +4,7 @@ from time import perf_counter
 
 import gradio as gr
 from jinja2 import Environment, FileSystemLoader
+import requests
 
 from backend.query_llm import generate
 from backend.semantic_search import retriever
@@ -20,6 +21,26 @@ env = Environment(loader=FileSystemLoader(proj_dir / 'templates'))
 template = env.get_template('template.j2')
 template_html = env.get_template('template_html.j2')
 
+def check_endpoint_status():
+    # Replace with the actual API URL and headers
+    api_url = os.getenv("ENDPOINT_URL")
+    headers = {
+        'accept': 'application/json',
+        'Authorization': f'Bearer {os.getenv("BEARER")}'
+        }
+
+    try:
+        response = requests.get(api_url, headers=headers)
+        response.raise_for_status()  # will throw an exception for non-200 status
+        data = response.json()
+
+        # Extracting the status information
+        status = data.get('status', {}).get('state', 'No status found')
+        message = data.get('status', {}).get('message', 'No message found')
+
+        return f"Status: {status}\nMessage: {message}"
+    except requests.exceptions.RequestException as e:
+        return f"Failed to get status: {str(e)}"
 
 def add_text(history, text):
     history = [] if history is None else history
@@ -51,6 +72,7 @@ def bot(history, system_prompt=""):
 
 with gr.Blocks() as demo:
     with gr.Tab("Application"):
+        output = gr.Textbox(check_endpoint_status, label="Endpoint Status (send chat to wake up)", every=1)
         chatbot = gr.Chatbot(
                 [],
                 elem_id="chatbot",