Create self_update_test.py

self_update_test.py

@@ -0,0 +1,56 @@
+
+# self_update_test.py
+import os
+import requests
+import time
+
+def test_self_update():
+    print("Testing self-update capability...")
+    
+    # Original response handler content
+    original_content = """def generate_response(prompt, tokenizer, model):
+    inputs = tokenizer(prompt, return_tensors="pt")
+    outputs = model.generate(**inputs, max_length=200)
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)"""
+    
+    # Updated version
+    updated_content = """def generate_response(prompt, tokenizer, model):
+    # Improved version with better parameters
+    inputs = tokenizer(prompt, return_tensors="pt")
+    outputs = model.generate(
+        **inputs, 
+        max_length=300,
+        temperature=0.7,
+        top_p=0.9,
+        repetition_penalty=1.1
+    )
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)"""
+    
+    # Save original version
+    with open("response_handler.py", "w") as f:
+        f.write(original_content)
+    
+    # Start the app (in a real test, this would be a separate process)
+    print("Starting with original version...")
+    
+    # Simulate update process
+    print("Triggering update...")
+    response = requests.post("http://localhost:7860/api/predict", json={"data": ["/update"]})
+    print(f"Update result: {response.json()['data']}")
+    
+    # Verify update
+    time.sleep(2)
+    with open("response_handler.py", "r") as f:
+        current_content = f.read()
+    
+    if updated_content in current_content:
+        print("✅ Self-update successful!")
+    else:
+        print("❌ Self-update failed")
+    
+    # Restore original
+    with open("response_handler.py", "w") as f:
+        f.write(original_content)
+
+if __name__ == "__main__":
+    test_self_update()