Update app.py

app.py

@@ -24,7 +24,7 @@ except Exception as e:
 
 def predict(input_text):
     if classifier is None:
-        return {"Error": "Model not loaded properly."}
+        return "Model not loaded properly.", {"Error": "Model not loaded properly."}
 
     try:
         # Predict the category
@@ -37,14 +37,20 @@ def predict(input_text):
         output = data["output"]
         metadata = data["metadata"]
 
-        return {
+        # Create a simple result string
+        simple_result = f"Category: {label}\nConfidence: {score:.2f}\nResult: {output}"
+
+        # Create the full JSON output
+        detailed_result = {
             "Category": label,
             "Confidence": score,
             "Output (Result)": output,
             "Metadata": metadata,
         }
+
+        return simple_result, detailed_result
     except Exception as e:
-        return {"Error": str(e)}
+        return "An error occurred.", {"Error": str(e)}
 
 # Gradio interface
 import gradio as gr
@@ -52,7 +58,10 @@ import gradio as gr
 interface = gr.Interface(
     fn=predict,
     inputs=gr.Textbox(lines=2, placeholder="Enter a math problem..."),
-    outputs="json",
+    outputs=[
+        gr.Textbox(label="Simple Output"),  # Display only the result
+        gr.JSON(label="Detailed JSON Output"),  # Display full JSON
+    ],
     title="Sangapac Math Model",
     description="A model to classify math problems into categories like Arithmetic, Multiplication, Division, Algebra, and Geometry.",
 )