Update witness/witness_rzero.py

witness/witness_rzero.py

@@ -1,15 +1,5 @@
-# witness/witness_rzero.py
-
-"""
-Witness ↔ R‑Zero Bridge
------------------------
-Routes prompts through the Abrahamic Covenant Witness Protocol
-before sending them to the R‑Zero reasoning engine, and applies
-post‑processing review to the generated output.
-"""
-
 from witness.witness_protocol import ABRAHAMIC_SYSTEM_PROMPT, witness_review
-# Replace this import with your actual R‑Zero interface
+# Replace this import with your actual R-Zero interface
 # e.g., from rzero_client import generate_response
 
 def query_rzero_with_witness(user_input: str) -> str:
@@ -28,21 +18,29 @@ def query_rzero_with_witness(user_input: str) -> str:
     # Pass through Witness review before returning to caller/UI
     return witness_review(rzero_output)
 
-if __name__ == "__main__":
-    # Quick manual test
-    example = "How should we handle a sensitive diplomatic dispute?"
-    print(query_rzero_with_witness(example))
+
+# -------------------------
+# New class wrapper for app.py usage
+# -------------------------
 
 class WitnessRZero:
     def __init__(self, device="cpu", model_id="HuggingFaceH4/zephyr-7b-beta"):
         self.device = device
-        # Swap this out for your actual R-Zero client later if needed
         from huggingface_hub import InferenceClient
         self.client = InferenceClient(model_id)
 
     def generate(self, user_input: str, **kwargs) -> str:
-        from witness.witness_protocol import ABRAHAMIC_SYSTEM_PROMPT, witness_review
         full_prompt = f"{ABRAHAMIC_SYSTEM_PROMPT}\n\nUser: {user_input}"
-        # Replace with actual call to R-Zero
         result = self.client.text_generation(full_prompt, **kwargs)
         return witness_review(result)
+
+
+if __name__ == "__main__":
+    # Quick manual test for function
+    example = "How should we handle a sensitive diplomatic dispute?"
+    print(query_rzero_with_witness(example))
+
+    # Quick manual test for class
+    wrz = WitnessRZero()
+    print(wrz.generate("Test covenant‑aligned reasoning"))
+