Update app.py

app.py

@@ -37,8 +37,19 @@ def predict(input_text):
         output = data["output"]
         metadata = data["metadata"]
 
+        # Extract metadata details
+        difficulty = metadata.get("difficulty", "Unknown")
+        steps = metadata.get("steps", ["No steps available"])
+
         # Create a simple result string
-        simple_result = f"Category: {label}\nConfidence: {score:.2f}\nResult: {output}"
+        steps_text = "\n".join(f"- {step}" for step in steps)
+        simple_result = (
+            f"Category: {label}\n"
+            f"Confidence: {score:.2f}\n"
+            f"Result: {output}\n"
+            f"Difficulty: {difficulty}\n"
+            f"Steps:\n{steps_text}"
+        )
 
         # Create the full JSON output
         detailed_result = {