Adding rich text box

app.py

@@ -4,6 +4,7 @@ from pathlib import Path
 from time import perf_counter
 
 import gradio as gr
+from gradio_rich_textbox import RichTextbox
 from jinja2 import Environment, FileSystemLoader
 from transformers import AutoTokenizer
 
@@ -103,7 +104,7 @@ def process_example(text, history=[]):
 
 with gr.Blocks() as demo:
     gr.Markdown(intro_md)
-    endpoint_status = gr.Textbox(check_endpoint_status, label="Inference Endpoint Status", every=1)
+    endpoint_status = RichTextbox(check_endpoint_status, label="Inference Endpoint Status", every=1)
     with gr.Tab("Arabic-RAG"):
         chatbot = gr.Chatbot(
                 [],

backend/query_llm.py

@@ -65,6 +65,9 @@ def check_endpoint_status():
         status = data.get('status', {}).get('state', 'No status found')
         message = data.get('status', {}).get('message', 'No message found')
 
-        return f"Status: {status}\nMessage: {message}"
+        if status == "scaledToZero":
+            return f"<div style='color: red; font-size: 20px; font-weight: bold;'>Status: {status}<br>Message: {message}</div>"
+        else:
+            return f"<div>Status: {status}<br>Message: {message}</div>"
     except requests.exceptions.RequestException as e:
-        return f"Failed to get status: {str(e)}"
+        return f"<div>Failed to get status: {str(e)}</div>"

requirements.txt

@@ -8,4 +8,5 @@ farm-haystack[inference]
 ipywidgets==8.1.1
 plotly==5.18.0
 lancedb==0.3.2
-bitsandbytes==0.41.1
+bitsandbytes==0.41.1
+./gradio_packages/gradio_rich_textbox-0.2.0-py3-none-any.whl