Add updated files

alternatives.py

@@ -0,0 +1,18 @@
+ALTERNATIVES = {
+    "simple factual question": [
+        "Use a traditional Search Engine (like Google Search)",
+        "Check Wikipedia or an offline encyclopedia",
+        "Consult a subject-matter expert or ask a friend"
+    ],
+    "mathematical calculation": [
+        "Use a dedicated math engine like Wolfram Alpha",
+        "Fire up Geogebra or a standard calculator app",
+        "Solve it on good old-fashioned paper"
+    ],
+    "creative content generation": [
+        "Brainstorm with a notebook and pen",
+        "Use offline writing prompts or a dictionary/thesaurus",
+        "Listen to a podcast or read a book for inspiration"
+    ],
+    "complex research query": [] 
+}

app.py

@@ -1,7 +1,48 @@
-import gradio as gr
+from transformers import pipeline
+from game_data import calculate_impact
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Load zero-shot classification pipeline 
+# (This runs server-side and uses the free CPU tier effectively)
+print("Loading classification model (facebook/bart-large-mnli)...")
+classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+# Predefined categories mapping to your zero-shot labels
+CATEGORIES = [
+    "simple factual question",
+    "mathematical calculation",
+    "creative content generation",
+    "complex research query"
+]
+
+def classify_query(user_input):
+    """
+    Takes user input, classifies it using zero-shot classification, 
+    and returns the highest probability category and its confidence score.
+    """
+    result = classifier(user_input, CATEGORIES)
+    top_category = result['labels'][0]
+    confidence_score = result['scores'][0]
+    
+    return top_category, confidence_score
+
+# --- Testing Block (You can remove this when building the Gradio UI) ---
+if __name__ == "__main__":
+    # Test 1: Simple Fact
+    test_query = "What is the capital of France?"
+    print(f"\nQuery: '{test_query}'")
+    
+    category, confidence = classify_query(test_query)
+    print(f"Classification: {category} (Confidence: {confidence:.2f})")
+    
+    impact = calculate_impact(category, confidence, test_query)
+    print(f"Impact: {impact['water']}L water, {impact['energy']}kWh energy")
+
+    # Test 2: Creative Generation
+    test_query_2 = "Write a 500-word sci-fi story about a robot who learns to love painting landscapes."
+    print(f"\nQuery: '{test_query_2}'")
+    
+    category_2, confidence_2 = classify_query(test_query_2)
+    print(f"Classification: {category_2} (Confidence: {confidence_2:.2f})")
+    
+    impact_2 = calculate_impact(category_2, confidence_2, test_query_2)
+    print(f"Impact: {impact_2['water']}L water, {impact_2['energy']}kWh energy")

game_data.py

@@ -0,0 +1,34 @@
+# Base costs mapping: {"category": {"water": L, "energy": kWh}}
+BASE_COSTS = {
+    "simple factual question": {"water": 0.5, "energy": 0.01},
+    "mathematical calculation": {"water": 0.3, "energy": 0.005},
+    "creative content generation": {"water": 5.0, "energy": 0.5},
+    "complex research query": {"water": 10.0, "energy": 1.0}
+}
+
+def get_confidence_multiplier(confidence_score):
+    if confidence_score > 0.8:
+        return 1.0
+    elif 0.5 <= confidence_score <= 0.8:
+        return 1.2  # Add 20% uncertainty buffer
+    else:
+        return 1.5  # Add 50% buffer
+
+def get_length_factor(query_text):
+    word_count = len(query_text.split())
+    if word_count < 20:
+        return 1.0
+    elif 20 <= word_count <= 50:
+        return 1.3
+    else:
+        return 1.5
+
+def calculate_impact(category, confidence_score, query_text):
+    base_cost = BASE_COSTS.get(category, {"water": 0, "energy": 0})
+    conf_mult = get_confidence_multiplier(confidence_score)
+    len_factor = get_length_factor(query_text)
+
+    water_impact = base_cost["water"] * conf_mult * len_factor
+    energy_impact = base_cost["energy"] * conf_mult * len_factor
+
+    return {"water": round(water_impact, 2), "energy": round(energy_impact, 3)}