Upload app.py with huggingface_hub

app.py

@@ -59,35 +59,38 @@ def classify_text(text):
     return output
 
 # Example questions - mix of actual ESS data and generated diverse questions
-examples = [
-    # EDUCATION (most common - 146 samples)
+# Using exact category names from pydantic_classes.py
+import random
+
+all_examples = [
+    # EDUCATION
     ["What is the highest level of education you have successfully completed?"],
     ["What is the highest level of education your mother successfully completed?"],
     ["How many years of full-time education have you completed?"],
 
-    # POLITICS (100 samples)
+    # POLITICS
     ["Which party did you vote for in the last national election?"],
     ["Trust in country's parliament"],
     ["How satisfied are you with the way democracy works in your country?"],
     ["How much do you trust the legal system?"],
 
-    # HEALTH (90 samples)
+    # HEALTH
     ["How satisfied are you with the healthcare system?"],
     ["Which health problems that you had in the last 12 months hampered you in your daily activities?"],
     ["How is your health in general - very good, good, fair, bad, or very bad?"],
 
-    # LABOUR AND EMPLOYMENT (82 samples)
+    # LABOUR AND EMPLOYMENT
     ["What best describes what you have been doing for the last 7 days - in paid work?"],
     ["Which description best describes the sort of work your mother did when you were 14?"],
     ["How many hours do you normally work per week in your main job?"],
     ["Are you a member of a trade union or similar organization?"],
 
-    # SOCIETY AND CULTURE (73 samples)
+    # SOCIETY AND CULTURE
     ["How often do you pray apart from at religious services?"],
     ["How important is it to always behave properly and avoid doing anything people would say is wrong?"],
     ["Do you consider yourself as belonging to any particular religion or denomination?"],
 
-    # DEMOGRAPHY
+    # DEMOGRAPHY (POPULATION, VITAL STATISTICS, AND CENSUSES)
     ["What is your age?"],
     ["What is your gender?"],
     ["What is your current legal marital status?"],
@@ -122,19 +125,35 @@ examples = [
     ["Do you rent or own your accommodation?"],
     ["How many rooms do you have for your household's use only?"],
 
-    # SOCIAL WELFARE
+    # SOCIAL WELFARE POLICY AND SYSTEMS
     ["Should the government reduce income differences?"],
     ["How satisfied are you with the state of social benefits in your country?"],
 
-    # TRANSPORT
+    # TRANSPORT AND TRAVEL
     ["How long does your daily commute to work take?"],
     ["What is your main mode of transportation?"],
 
     # SCIENCE AND TECHNOLOGY
     ["To what extent do you think scientific advances benefit society?"],
     ["How often do you use a smartphone or tablet?"],
+
+    # HISTORY
+    ["What do you think about your country's colonial past?"],
+    ["How important is it to preserve historical monuments?"],
+
+    # TRADE, INDUSTRY AND MARKETS
+    ["Do you work in the private or public sector?"],
+    ["How do you feel about free trade agreements?"],
+
+    # OTHER
+    ["What are your thoughts on the future?"],
+    ["How do you define quality of life?"],
 ]
 
+# Shuffle and select 30 examples
+random.seed(42)  # For reproducibility
+examples = random.sample(all_examples, min(30, len(all_examples)))
+
 # Custom CSS for Sikt branding using design tokens
 custom_css = """
 :root {
@@ -271,26 +290,26 @@ demo = gr.Interface(
             categorizing them into subject areas. The model was trained on European Social Survey metadata
             and can classify questions into <strong>19 categories</strong>:
         </p>
-        <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(250px, 1fr)); gap: 0.5rem; margin-top: 1rem;">
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">📚 Education</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">🏛️ Politics</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">🏥 Health</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">💼 Labour & Employment</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">🌍 Society & Culture</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">💰 Economics</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">🧠 Psychology</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">👥 Demography</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">🏠 Housing & Land Use</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">🌱 Natural Environment</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">⚖️ Law, Crime & Legal Systems</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">📺 Media, Communication & Language</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">👔 Social Stratification & Groupings</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">🤝 Social Welfare Policy & Systems</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">🚗 Transport & Travel</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">🏪 Trade, Industry & Markets</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">🔬 Science & Technology</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">📜 History</span>
-            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.85rem; border-left: 3px solid #7d5da6;">📋 Other</span>
+        <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(280px, 1fr)); gap: 0.5rem; margin-top: 1rem;">
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">📚 EDUCATION</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">🏛️ POLITICS</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">🏥 HEALTH</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">💼 LABOUR AND EMPLOYMENT</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">🌍 SOCIETY AND CULTURE</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">💰 ECONOMICS</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">🧠 PSYCHOLOGY</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">👥 DEMOGRAPHY (POPULATION, VITAL STATISTICS, AND CENSUSES)</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">🏠 HOUSING AND LAND USE</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">🌱 NATURAL ENVIRONMENT</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">⚖️ LAW, CRIME AND LEGAL SYSTEMS</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">📺 MEDIA, COMMUNICATION AND LANGUAGE</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">👔 SOCIAL STRATIFICATION AND GROUPINGS</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">🤝 SOCIAL WELFARE POLICY AND SYSTEMS</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">🚗 TRANSPORT AND TRAVEL</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">🏪 TRADE, INDUSTRY AND MARKETS</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">🔬 SCIENCE AND TECHNOLOGY</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">📜 HISTORY</span>
+            <span style="padding: 0.5rem; background: white; border-radius: 4px; font-size: 0.8rem; border-left: 3px solid #7d5da6;">📋 OTHER</span>
         </div>
     </div>