add test framework and skin change on flag

.gitignore

@@ -2,4 +2,9 @@ myenv/
 .DS_Store
 .env
 
-__pycache__/
+__pycache__/
+.pytest_cache/
+
+report.html
+style.css
+screenshots/

app.py

@@ -3,6 +3,8 @@ from transformers import pipeline
 from ldclient import LDClient, Config, Context
 import os
 
+style = False
+
 # Retrieve the LaunchDarkly SDK key from environment variables
 ld_sdk_key = os.getenv("LAUNCHDARKLY_SDK_KEY")
 
@@ -21,6 +23,13 @@ def get_model_config(user_name):
     model_id = flag_variation.get("modelID", "distilbert-base-uncased")
     return model_id
 
+# Function to get Style from LaunchDarkly
+def get_style_config():
+    flag_key = "style"
+    style_context = Context.builder("context-key-style").build()
+    flag_variation = ld_client.variation(flag_key, style_context,default=False)
+    return flag_variation
+
 # Function to translate sentiment labels to user-friendly terms
 def translate_label(label):
     label_mapping = {
@@ -35,6 +44,26 @@ def translate_label(label):
     }
     return label_mapping.get(label, "Unknown")
 
+style = get_style_config()
+
+# popup with the styel value
+st.write(f"Style: {style}")
+
+if style:
+    custom_css = """
+        <style>
+            html, body {
+                height: 100%;
+            }
+            .main{
+                background: green;
+            }
+        </style>
+        """
+    st.markdown(custom_css, unsafe_allow_html=True)
+else:
+    cust_css = ""
+    
 # Streamlit app
 st.title("Sentiment Analysis Demo with AI Model Flags")
 

requirements.txt

@@ -4,4 +4,6 @@ launchdarkly-server-sdk
 torch
 Flask
 datasets
-gunicorn
+gunicorn
+pytest-playwright
+pytest-html

test/test_chatbot.py

@@ -0,0 +1,27 @@
+import re
+from playwright.sync_api import Page, expect, Browser
+
+def test_has_title(page: Page):
+    page.goto("http://localhost:8501")
+
+    # Expect a title "to contain" a substring.
+    expect(page).to_have_title(re.compile("app"))
+
+def test_get_started_link(browser: Browser):
+
+        for _ in range(30):
+            # create a new incognito browser context
+            context = browser.new_context()
+            # create a new page inside context.
+            page = context.new_page()
+            # clear cache
+            page.context.clear_cookies()
+            page.goto("http://localhost:8501")
+
+            # expect Sentiment Analysis Demo with AI Model Flags
+            locator = page.locator('.main')
+            expect(locator).to_contain_text("Sentiment Analysis Demo with AI Model Flags")
+            # screenshot
+            page.screenshot(path=f"test_get_started_link{_}.png")
+            # dispose context once it is no longer needed.
+            context.close()