Spaces:
Sleeping
Sleeping
| import requests | |
| import base64 | |
| import os | |
| API_URL = "http://localhost:8001/api/voice-detection" | |
| API_KEY = "secret123" | |
| def test_api(): | |
| print(f"Testing API at {API_URL}...") | |
| # Create a dummy MP3 file (1 frame of silence) or just random bytes with mp3 header if possible | |
| # For simplicity, we'll assume the decoder can handle a small valid MP3 or we use a proper base64 string | |
| # This is a minimal valid MP3 frame (MPEG 1 Layer 3, 128kbps, 44.1kHz) | |
| dummy_mp3_hex = "FFFB906400000000000000000000000000000000" | |
| # That hex is likely too short/invalid, but let's try a very basic string | |
| # Note: The decoder in app/audio/decoder.py just writes bytes. librosa.load might fail if invalid. | |
| # Ideally we need a real file. | |
| # Using a placeholder valid base64 for a silent mp3 (extremely short) | |
| # Read a real file from the dataset | |
| try: | |
| # Try to find a real file | |
| import glob | |
| files = glob.glob("dataset/human/**/*.wav", recursive=True) | |
| if not files: | |
| files = glob.glob("dataset/human/**/*.mp3", recursive=True) | |
| if files: | |
| file_path = files[0] | |
| print(f"Using audio file: {file_path}") | |
| with open(file_path, "rb") as f: | |
| audio_data = f.read() | |
| valid_mp3_b64 = base64.b64encode(audio_data).decode("utf-8") | |
| else: | |
| print("No dataset files found, falling back to dummy (might fail)") | |
| valid_mp3_b64 = "/+MYxAAAAANIAAAAAExBTUUzLjEwMKqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqg=" | |
| except Exception as e: | |
| print(f"Error reading file: {e}") | |
| valid_mp3_b64 = "" | |
| payload = { | |
| "language": "Tamil", | |
| "audioFormat": "mp3", # The API might ignore this if decoder handles headers, but we send what we have | |
| "audioBase64": valid_mp3_b64 | |
| } | |
| headers = { | |
| "x-api-key": API_KEY, | |
| "Content-Type": "application/json" | |
| } | |
| try: | |
| response = requests.post(API_URL, json=payload, headers=headers) | |
| print(f"Status Code: {response.status_code}") | |
| print(f"Response: {response.text}") | |
| if response.status_code == 200: | |
| data = response.json() | |
| print("✅ API Request Successful") | |
| print(f" Classification: {data['classification']}") | |
| print(f" Confidence: {data['confidenceScore']}") | |
| print(f" Explanation: {data['explanation']}") | |
| else: | |
| print("❌ API Request Failed") | |
| except Exception as e: | |
| print(f"❌ Connection Failed: {e}") | |
| print("Ensure the server is running: uvicorn app.main:app --reload") | |
| if __name__ == "__main__": | |
| test_api() | |