Update app.py

app.py

@@ -14,12 +14,6 @@ st.set_page_config(
     menu_items=None
 )
 
-#st.logo(
-#    "https://dejan.ai/wp-content/uploads/2024/02/dejan-300x103.png",
-#    size="large",
-#    link="https://dejan.ai/"
-#)
-
 MODEL_ID = "dejanseo/query-grounding"
 HF_TOKEN = os.getenv("HF_TOKEN")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -38,33 +32,52 @@ def classify(prompt: str):
         confidence = probs[pred].item()
         return pred, confidence
 
-# Hide HF space bar + Streamlit chrome
+# Font and style overrides
 st.markdown("""
     <style>
         @import url('https://fonts.googleapis.com/css2?family=Montserrat:wght@400;600&display=swap');
 
-        html, body, div, span, input, label, textarea, button, h1, h2, p,table {
+        html, body, div, span, input, label, textarea, button, h1, h2, p, table {
             font-family: 'Montserrat', sans-serif !important;
         }
 
-        /* Streamlit's dynamic classes */
         [class^="css-"], [class*=" css-"] {
             font-family: 'Montserrat', sans-serif !important;
         }
 
         header {visibility: hidden;}
-        
     </style>
 """, unsafe_allow_html=True)
 
-
+# UI
 st.title("QDG Classifier")
 st.write("Built by [**Dejan AI**](https://dejan.ai)")
 st.write("Google's AI models can request grounding with search to deliver more accurate or up-to-date answers. This tool predicts whether a search query, question, or prompt is likely to be grounded.")
-user_input = st.text_area("Enter one search query, question, or prompt per line:")
+
+# Placeholder example prompts
+example_text = """how would a cat describe a dog
+how to reset a Nest thermostat
+write a poem about time
+is there a train strike in London today
+summarize the theory of relativity
+who won the champions league last year
+explain quantum computing to a child
+weather in tokyo tomorrow
+generate a social media post for Earth Day
+what is the latest iPhone model"""
+
+user_input = st.text_area(
+    "Enter one search query, question, or prompt per line:",
+    placeholder=example_text
+)
 
 if st.button("Classify"):
-    prompts = [line.strip() for line in user_input.strip().split("\n") if line.strip()]
+    raw_input = user_input.strip()
+    if raw_input:
+        prompts = [line.strip() for line in raw_input.split("\n") if line.strip()]
+    else:
+        prompts = [line.strip() for line in example_text.split("\n")]
+
     if not prompts:
         st.warning("Please enter at least one prompt.")
     else:
@@ -93,4 +106,8 @@ if st.button("Classify"):
                     },
                     hide_index=True,
                 )
-        info_box.empty()
+        info_box.empty()
+
+        # Promo message shown only after results
+        st.subheader("Working together.")
+        st.write("[**Schedule a call**](https://dejan.ai/call/) to see how we can help you.")