Add updated files

app.py

@@ -1,3 +1,4 @@
+import gradio as gr
 from transformers import pipeline
 from game_data import calculate_impact
 
@@ -25,24 +26,32 @@ def classify_query(user_input):
     
     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})")
+def process_query(user_input):
+    """Processes the query for the Gradio interface."""
+    category, confidence = classify_query(user_input)
+    impact = calculate_impact(category, confidence, user_input)
     
-    impact = calculate_impact(category, confidence, test_query)
-    print(f"Impact: {impact['water']}L water, {impact['energy']}kWh energy")
+    result_text = (
+        f"**Category:** {category}\n"
+        f"**Confidence:** {confidence:.2f}\n"
+        f"**Water Wasted:** {impact['water']}L\n"
+        f"**Energy Wasted:** {impact['energy']}kWh"
+    )
+    return result_text
 
-    # 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}'")
+# Create a basic Gradio interface to keep the Space alive
+with gr.Blocks() as demo:
+    gr.Markdown("# 💧 EcoQueryQuest (Test UI)")
+    gr.Markdown("Type a query to see its hidden environmental cost!")
+    
+    with gr.Row():
+        input_box = gr.Textbox(label="Your Query", placeholder="e.g., What is the capital of France?")
     
-    category_2, confidence_2 = classify_query(test_query_2)
-    print(f"Classification: {category_2} (Confidence: {confidence_2:.2f})")
+    submit_btn = gr.Button("Analyze")
+    output_box = gr.Markdown(label="Impact Results")
     
-    impact_2 = calculate_impact(category_2, confidence_2, test_query_2)
-    print(f"Impact: {impact_2['water']}L water, {impact_2['energy']}kWh energy")
+    submit_btn.click(fn=process_query, inputs=input_box, outputs=output_box)
+
+# Launch the app so it stays running on Hugging Face
+if __name__ == "__main__":
+    demo.launch()