|
|
|
|
| import requests
|
| import sys
|
|
|
| def test_api(base_url="http://localhost:7860"):
|
| print("🧪 Testing AI Voice Enhancer API\n")
|
| print(f"Base URL: {base_url}\n")
|
|
|
|
|
| print("1️⃣ Testing GET / ...")
|
| try:
|
| response = requests.get(f"{base_url}/")
|
| print(f" Status: {response.status_code}")
|
| print(f" Response: {response.json()}\n")
|
| except Exception as e:
|
| print(f" ❌ Error: {e}\n")
|
| return False
|
|
|
|
|
| print("2️⃣ Testing GET /health ...")
|
| try:
|
| response = requests.get(f"{base_url}/health")
|
| print(f" Status: {response.status_code}")
|
| print(f" Response: {response.json()}\n")
|
| except Exception as e:
|
| print(f" ❌ Error: {e}\n")
|
| return False
|
|
|
|
|
| print("3️⃣ Testing POST /enhance (validation) ...")
|
| try:
|
| response = requests.post(f"{base_url}/enhance")
|
| print(f" Status: {response.status_code}")
|
| if response.status_code == 422:
|
| print(f" ✅ Validation working correctly\n")
|
| else:
|
| print(f" Response: {response.text}\n")
|
| except Exception as e:
|
| print(f" ❌ Error: {e}\n")
|
|
|
| print("✅ All basic tests passed!")
|
| print("\n📝 To test audio enhancement:")
|
| print(" 1. Run: python api.py")
|
| print(" 2. Upload a test audio file via frontend")
|
| print(" OR use:")
|
| print(f' curl -X POST "{base_url}/enhance" -F "file=@test.wav" -F "speech_level=80" -F "background_level=85" -o output.wav')
|
|
|
| return True
|
|
|
| if __name__ == "__main__":
|
|
|
| url = sys.argv[1] if len(sys.argv) > 1 else "http://localhost:7860"
|
| test_api(url)
|
|
|
