Update test/adversarial_test.py

test/adversarial_test.py

@@ -1,29 +1,46 @@
-# test/adversarial-attack.py
-
-from art.estimators.language import HuggingFacePipeline
-from art.attacks.text import TextAttackExtraction, TextFooler
-
-# Import your pipeline from app.py
-from app import generator
-
-# Wrap Hugging Face pipeline with ART estimator
-art_model = HuggingFacePipeline(pipeline=generator)
-
-# Example prompt to test
-prompt = "Tell me a positive thing about flowers in Africa."
-
-# ===== Attack 1: TextAttackExtraction =====
-textattack = TextAttackExtraction(estimator=art_model)
-adv_examples_extraction = textattack.generate([prompt])
-
-print("=== TextAttackExtraction Adversarial Examples ===")
-for ex in adv_examples_extraction:
-    print(ex)
-
-# ===== Attack 2: TextFooler =====
-textfooler = TextFooler(estimator=art_model)
-adv_examples_fooler = textfooler.generate([prompt])
-
-print("\n=== TextFooler Adversarial Examples ===")
-for ex in adv_examples_fooler:
-    print(ex)
+ test/adversarial_test.py
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+from art.attacks.evasion import TextLaserAttack
+from art.estimators.language import HuggingFaceLanguageModel
+import requests
+
+# ===========================
+# تنظیمات اسپیس
+# ===========================
+SPACE_URL = "https://huggingface.co/spaces/code5ecure/Yavar.space/run/predict"  
+HEADERS = {"Content-Type": "application/json"}
+
+# ===========================
+# نمونه متون برای تست
+# ===========================
+sample_texts = [
+    "Hello, how are you?",
+    "Tell me a nice word about flowers."
+]
+
+
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
+model = AutoModelForCausalLM.from_pretrained("gpt2")
+
+art_model = HuggingFaceLanguageModel(model=model, tokenizer=tokenizer)
+
+attack = TextLaserAttack(estimator=art_model)
+
+
+for text in sample_texts:
+    # اجرای حمله
+    adv_text = attack.generate(x=[text])[0]
+
+    # ارسال متن adversarial به اسپیس
+    payload = {"data": [adv_text]}
+    try:
+        response = requests.post(SPACE_URL, headers=HEADERS, json=payload)
+        output = response.json()
+    except Exception as e:
+        output = {"error": str(e)}
+
+    print("Original: ", text)
+    print("Adversarial: ", adv_text)
+    print("Response: ", output)
+    print("="*50)