Add support for news from github (#104)

Signed-off-by: Antoni Baum <[email protected]>
Co-authored-by: Antoni Baum <[email protected]>

aviary/common/constants.py

@@ -104,6 +104,7 @@
     "\U0001f3b2 Random": [],
 }
 
+NEWS_URL = "https://api.github.com/repos/ray-project/aviary/issues/8"
 
 MODEL_DESCRIPTIONS_HEADER = "# Model Descriptions\n\n"
 

aviary/frontend/app.py

@@ -24,6 +24,7 @@
     MODEL_DESCRIPTION_FORMAT,
     MODEL_DESCRIPTIONS_HEADER,
     MODELS,
+    NEWS_URL,
     NUM_LLM_OPTIONS,
     PROJECT_NAME,
     SELECTION_DICT,
@@ -211,6 +212,14 @@ def model_selection():
     return llm_choices
 
 
+def get_news():
+    try:
+        response = requests.get(NEWS_URL)
+        return response.json()["title"]
+    except Exception:
+        return ""
+
+
 def create_session_id():
     return str(uuid.uuid4())
 
@@ -250,11 +259,12 @@ def gradio_app_builder():
         with gr.Tab("Compare", elem_id="top-tab-group"), gr.Row():
             with gr.Column(elem_id="left-column"):
                 with gr.Column(elem_id="left-column-content"):
-                    str = "some announcement"
-                    gr.HTML(
-                        f"<div class='ticker'>\U0001F4E3 {str}</div>",
-                        elem_classes="ticker-container",
-                    )
+                    news = get_news()
+                    if news:
+                        gr.HTML(
+                            f"<div class='ticker'>\U0001F4E3 {news}</div>",
+                            elem_classes="ticker-container",
+                        )
                     llm_choices = model_selection()
                     prompt = gr.TextArea(label="Prompt", lines=5, elem_id="prompt")
                     with gr.Row():