Initial commit

app.py

@@ -0,0 +1,111 @@
+import gradio as gr
+from transformers import pipeline
+
+# Load the toxicity classification pipeline
+print("Loading toxicity classifier pipeline...")
+toxicity_pipeline = pipeline(
+    "text-classification",
+    model="s-nlp/roberta_toxicity_classifier",
+    tokenizer="s-nlp/roberta_toxicity_classifier",
+    # return_all_scores=True  # This ensures we get scores for both classes
+)
+print("Pipeline loaded successfully!")
+
+def toxicity_classification(text: str) -> dict:
+    """
+    Classify the toxicity of the given text.
+
+    Args:
+        text (str): The text to analyze
+
+    Returns:
+        dict: A dictionary containing toxicity scores and classification
+    """
+    if not text.strip():
+        return {
+            "error": "Please enter some text to analyze"
+        }
+    
+    try:
+        # Get predictions using the pipeline
+        result = toxicity_pipeline(text)[0]
+        
+        # The pipeline returns a list like: [{'label': 'LABEL_0', 'score': 0.8}]
+        # LABEL_0 is non-toxic, LABEL_1 is toxic (based on the model description)
+        
+        # Convert to a more readable format
+
+        label = result['label']
+        confidence = result['score']
+
+        classification = "non-toxic" if label == "LABEL_0" else "toxic"
+        
+        return {
+            "classification": classification,
+            "confidence": round(confidence, 4)
+        }
+        
+    except Exception as e:
+        return {
+            "error": f"Error processing text: {str(e)}"
+        }
+
+# Create the Gradio interface
+demo = gr.Interface(
+    fn=toxicity_classification,
+    inputs=gr.Textbox(
+        placeholder="Enter text to analyze for toxicity...",
+        lines=3,
+        label="Input Text"
+    ),
+    outputs=gr.JSON(label="Toxicity Analysis Results"),
+    title="Text Toxicity Classification",
+    description="Analyze text toxicity using RoBERTa transformer model (s-nlp/roberta_toxicity_classifier)",
+    examples=[
+        ["You are amazing!"],
+        ["This is a wonderful day."],
+        ["I disagree with your opinion."],
+        ["The weather is nice today."]
+    ]
+)
+
+# Launch the interface and MCP server
+if __name__ == "__main__":
+    # Add debugging section for local testing
+    import sys
+    
+    # Check if running in debug mode (pass 'debug' as command line argument)
+    if len(sys.argv) > 1 and sys.argv[1] == "debug":
+        print("=" * 50)
+        print("DEBUG MODE - Testing toxicity classification")
+        print("=" * 50)
+        
+        # Test cases for debugging
+        test_cases = [
+            "You are amazing!",
+            "This is a wonderful day.",
+            "I hate you so much!",
+            "The weather is nice today.",
+            "You're such an idiot!",
+            "I disagree with your opinion.",
+            ""  # Empty string test
+        ]
+        
+        for i, test_text in enumerate(test_cases):
+            print(f"\n--- Test Case {i} ---")
+            print(f"Input: '{test_text}'")
+            
+            # Set breakpoint here for debugging
+            import pdb; pdb.set_trace()
+            
+            # Call the function
+            result = toxicity_classification(test_text)
+            
+            print(f"Output: {result}")
+            print("-" * 30)
+        
+        print("\nDebug testing completed!")
+        
+    else:
+        # Normal Gradio mode
+        demo.launch(mcp_server=True)

requirements.txt

@@ -0,0 +1,128 @@
+altgraph @ file:///AppleInternal/Library/BuildRoots/2c89a47b-9dd5-11ef-938f-6e654a286000/Library/Caches/com.apple.xbs/Sources/python3/altgraph-0.17.2-py2.py3-none-any.whl
+anyio==4.9.0
+appnope==0.1.4
+argon2-cffi==23.1.0
+argon2-cffi-bindings==21.2.0
+arrow==1.3.0
+asttokens==3.0.0
+async-lru==2.0.5
+attrs==25.3.0
+babel==2.17.0
+beautifulsoup4==4.13.3
+bleach==6.2.0
+bs4==0.0.2
+certifi==2025.1.31
+cffi==1.17.1
+charset-normalizer==3.4.1
+cloudpickle==3.1.1
+comm==0.2.2
+contourpy==1.3.0
+cycler==0.12.1
+debugpy==1.8.13
+decorator==5.2.1
+defusedxml==0.7.1
+exceptiongroup==1.2.2
+executing==2.2.0
+fastjsonschema==2.21.1
+filelock==3.17.0
+fonttools==4.56.0
+fqdn==1.5.1
+fsspec==2025.2.0
+future @ file:///AppleInternal/Library/BuildRoots/2c89a47b-9dd5-11ef-938f-6e654a286000/Library/Caches/com.apple.xbs/Sources/python3/future-0.18.2-py3-none-any.whl
+gym==0.26.2
+gym-notices==0.0.8
+h11==0.16.0
+httpcore==1.0.9
+httpx==0.28.1
+idna==3.10
+importlib_metadata==8.6.1
+importlib_resources==6.5.2
+ipykernel==6.29.5
+ipython==8.18.1
+isoduration==20.11.0
+jax==0.4.30
+jaxlib==0.4.30
+jedi==0.19.2
+Jinja2==3.1.5
+json5==0.12.0
+jsonpointer==3.0.0
+jsonschema==4.23.0
+jsonschema-specifications==2025.4.1
+jupyter-events==0.12.0
+jupyter-lsp==2.2.5
+jupyter_client==8.6.3
+jupyter_core==5.7.2
+jupyter_server==2.15.0
+jupyter_server_terminals==0.5.3
+jupyterlab==4.4.2
+jupyterlab_pygments==0.3.0
+jupyterlab_server==2.27.3
+kiwisolver==1.4.7
+macholib @ file:///AppleInternal/Library/BuildRoots/2c89a47b-9dd5-11ef-938f-6e654a286000/Library/Caches/com.apple.xbs/Sources/python3/macholib-1.15.2-py2.py3-none-any.whl
+MarkupSafe==3.0.2
+matplotlib==3.9.4
+matplotlib-inline==0.1.7
+mistune==3.1.3
+ml_dtypes==0.5.1
+mpmath==1.3.0
+nbclient==0.10.2
+nbconvert==7.16.6
+nbformat==5.10.4
+nest-asyncio==1.6.0
+networkx==3.2.1
+notebook==7.4.2
+notebook_shim==0.2.4
+numpy==2.0.2
+opt_einsum==3.4.0
+overrides==7.7.0
+packaging==24.2
+pandas==2.2.3
+pandocfilters==1.5.1
+parso==0.8.4
+pexpect==4.9.0
+pillow==11.1.0
+platformdirs==4.3.6
+prometheus_client==0.21.1
+prompt_toolkit==3.0.50
+psutil==7.0.0
+ptyprocess==0.7.0
+pure_eval==0.2.3
+pycparser==2.22
+pygame==2.6.1
+Pygments==2.19.1
+pyhumps==3.8.0
+pyparsing==3.2.1
+PyPFF @ git+https://github.com/pro-football-focus/pypff.git@ca2c4a53a63c20267a900065e9bd0d194db1a077
+python-dateutil==2.9.0.post0
+python-json-logger==3.3.0
+pytz==2025.1
+PyYAML==6.0.2
+pyzmq==26.2.1
+referencing==0.36.2
+requests==2.32.3
+rfc3339-validator==0.1.4
+rfc3986-validator==0.1.1
+rpds-py==0.24.0
+scipy==1.13.1
+Send2Trash==1.8.3
+six @ file:///AppleInternal/Library/BuildRoots/2c89a47b-9dd5-11ef-938f-6e654a286000/Library/Caches/com.apple.xbs/Sources/python3/six-1.15.0-py2.py3-none-any.whl
+sniffio==1.3.1
+soupsieve==2.6
+stack-data==0.6.3
+sympy==1.13.1
+terminado==0.18.1
+tinycss2==1.4.0
+tomli==2.2.1
+torch==2.6.0
+tornado==6.4.2
+traitlets==5.14.3
+types-python-dateutil==2.9.0.20241206
+typing_extensions==4.12.2
+tzdata==2025.1
+uri-template==1.3.0
+urllib3==2.3.0
+wcwidth==0.2.13
+webcolors==24.11.1
+webencodings==0.5.1
+websocket-client==1.8.0
+zipp==3.21.0