Update app.py

app.py

@@ -127,11 +127,9 @@ class ModelWrapper:
             thinking_answer_text = remainder.split("</transcript>")[-1]
             return thinking_answer_text
         except:
-            # If parsing fails, return the portion after the input
             input_length = len(message)
             return output_text[input_length:] if len(output_text) > input_length else "No response generated."
 
-# — instantiate your model —
 MODEL_NAME = "Qwen/Qwen3-0.6B"
 model = ModelWrapper(MODEL_NAME)
 
@@ -141,21 +139,17 @@ def compliance_check(rules_text, transcript_text, thinking):
         rules = [r for r in rules_text.split("\n") if r.strip()]
         inp = format_rules(rules) + format_transcript(transcript_text)
         
-        # Limit max tokens to prevent oversized responses
         out = model.get_response(inp, enable_thinking=thinking, max_new_tokens=256)
         
-        # Clean up any malformed output and ensure it's a string
         out = str(out).strip()
         if not out:
             out = "No response generated. Please try with different input."
 
-        # Ensure the response isn't too long for an HTTP response by checking byte length
-        max_bytes = 2500  # A more generous limit, in bytes
+        max_bytes = 2500  
         out_bytes = out.encode('utf-8')
 
         if len(out_bytes) > max_bytes:
-            # Truncate the byte string, then decode back to a string, ignoring errors
-            # This prevents cutting a multi-byte character in half
+
             truncated_bytes = out_bytes[:max_bytes]
             out = truncated_bytes.decode('utf-8', errors='ignore')
             out += "\n\n[Response truncated to prevent server errors]"
@@ -163,12 +157,11 @@ def compliance_check(rules_text, transcript_text, thinking):
         return out
         
     except Exception as e:
-        error_msg = f"Error: {str(e)[:200]}"  # Limit error message length
+        error_msg = f"Error: {str(e)[:200]}"  
         print(f"Full error: {e}")
         return error_msg
 
 
-# — build Gradio interface —
 demo = gr.Interface(
     fn=compliance_check,
     inputs=[