Upload app.py

app.py

@@ -357,34 +357,28 @@ def analyze_solution(question: str, solution: str):
 
     return final_verdict    
      
-
-
-def classify_solution_stream(question: str, solution: str, progress=gr.Progress()):
-    # always yield EXACTLY THREE outputs (classification, explanation, status)
+def classify_solution_stream(question: str, solution: str):
+    # always yield exactly 3 outputs: (classification, explanation, status)
     if not question.strip() or not solution.strip():
         yield "Please fill in both fields", "", "⚠️ Provide a question and a solution."
         return
 
-    # lazy init if needed
     if not models_ready():
         yield "⏳ Loading models…", "", "Booting models…"
         msg = load_model()
-        progress(0.1, desc="Loading models")
         if not models_ready():
             yield "Models not loaded", "", f"❌ {msg}"
             return
 
     try:
         # Stage 1 — conceptual
-        progress(0.2, desc="Stage 1: Conceptual check")
         yield "⏳ Working…", "Starting conceptual check…", "🔎 **Stage 1:** running classifier…"
         conceptual = run_conceptual_check(question, solution, classifier_model, classifier_tokenizer)
         conf = conceptual['probabilities'][conceptual['prediction']]
-        yield "⏳ Working…", f"Stage 1: model predicts **{conceptual['prediction']}** (confidence {conf:.2%}). Now checking calculations…", \
+        yield "⏳ Working…", f"Stage 1: **{conceptual['prediction']}** (p={conf:.2%}). Now checking calculations…", \
               f"✅ **Stage 1 done** — prediction: **{conceptual['prediction']}** (p={conf:.2%})."
 
         # Stage 2 — computational
-        progress(0.6, desc="Stage 2: Computational check")
         yield "⏳ Working…", "Running computational check…", "🧮 **Stage 2:** extracting & evaluating equations…"
         computational = run_computational_check(solution, gemma_model, gemma_tokenizer)
 
@@ -407,24 +401,22 @@ def classify_solution_stream(question: str, solution: str, progress=gr.Progress(
                 explanation = "All calculations are correct, but there appears to be a conceptual error in the logic or setup of the solution."
                 status = "🟨 **Final:** conceptual issue."
 
-        progress(1.0, desc="Done")
         yield classification, explanation, status
 
     except Exception as e:
         logger.exception("inference failed")
         yield "Runtime error", f"{type(e).__name__}: {e}", "❌ Exception during inference."
 
-        
-# Create Gradio interface (streaming UI)
+        # ---------------- UI: streaming, no progress bars ----------------
 with gr.Blocks(title="Math Solution Classifier", theme=gr.themes.Soft()) as app:
     gr.Markdown("# 🧮 Math Solution Classifier")
     gr.Markdown(
         "Classify math solutions as **correct**, **conceptually flawed**, or **computationally flawed**. "
-        "You’ll see live status updates as the two-stage pipeline runs."
+        "Live status updates appear below as the two-stage pipeline runs."
     )
 
     with gr.Row():
-        # ---------- Left: inputs ----------
+        # -------- Left: inputs --------
         with gr.Column(scale=1):
             question_input = gr.Textbox(
                 label="Math Question",
@@ -440,13 +432,13 @@ with gr.Blocks(title="Math Solution Classifier", theme=gr.themes.Soft()) as app:
                 classify_btn = gr.Button("Classify Solution", variant="primary")
                 clear_btn = gr.Button("Clear")
 
-        # ---------- Right: outputs ----------
+        # -------- Right: outputs --------
         with gr.Column(scale=1):
             classification_output = gr.Textbox(label="Classification", interactive=False)
             explanation_output   = gr.Textbox(label="Explanation",   interactive=False, lines=6)
             status_output        = gr.Markdown(value="*(idle)*")  # live stage updates
 
-    # ---------- Examples ----------
+    # -------- Examples --------
     gr.Examples(
         examples=[
             ["Solve for x: 2x + 5 = 13",
@@ -459,13 +451,13 @@ with gr.Blocks(title="Math Solution Classifier", theme=gr.themes.Soft()) as app:
         inputs=[question_input, solution_input],
     )
 
-    # ---------- Wiring ----------
+    # -------- Wiring --------
     classify_btn.click(
-        fn=classify_solution_stream,  # your generator
+        fn=classify_solution_stream,  # <- generator that yields (classification, explanation, status)
         inputs=[question_input, solution_input],
         outputs=[classification_output, explanation_output, status_output],
-        show_progress="minimal",      # single compact global bar
-        concurrency_limit=1,          # <- set per-event, not in app.queue()
+        show_progress=False,          # <- no Gradio progress bars
+        concurrency_limit=1,          # <- per-event limit (good for GPU)
     )
 
     clear_btn.click(
@@ -481,10 +473,11 @@ with gr.Blocks(title="Math Solution Classifier", theme=gr.themes.Soft()) as app:
         queue=False,
     )
 
-# Enable queue without deprecated args (or omit entirely if you like)
+# enable queue for streaming (no deprecated args)
 app.queue()
 
 if __name__ == "__main__":
     app.launch()
 
 
+