Spaces:
Sleeping
Sleeping
| import os | |
| import sys | |
| import time | |
| # Add the parent directory to the path so we can import our modules | |
| sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) | |
| from models.braille_translator import text_to_braille, get_braille_metadata | |
| def test_braille_translation(text): | |
| """ | |
| Test Braille translation on a given text. | |
| Args: | |
| text: Text to translate to Braille | |
| Returns: | |
| Dictionary with test results | |
| """ | |
| start_time = time.time() | |
| # Translate to Braille | |
| try: | |
| result = text_to_braille(text, use_context=True) | |
| success = result['success'] | |
| braille_text = result.get('formatted_braille', '') | |
| error = result.get('error', None) | |
| except Exception as e: | |
| success = False | |
| braille_text = '' | |
| error = str(e) | |
| end_time = time.time() | |
| # Get metadata | |
| metadata = get_braille_metadata(text) | |
| # Compile results | |
| test_results = { | |
| 'original_text': text, | |
| 'success': success, | |
| 'processing_time': end_time - start_time, | |
| 'braille_text': braille_text[:100] + '...' if len(braille_text) > 100 else braille_text, | |
| 'word_count': metadata['word_count'], | |
| 'character_count': metadata['character_count'], | |
| 'line_count': metadata['line_count'] | |
| } | |
| if not success: | |
| test_results['error'] = error | |
| return test_results | |
| def run_braille_tests(): | |
| """ | |
| Run tests on sample menu texts. | |
| Returns: | |
| List of test results | |
| """ | |
| # Sample menu texts | |
| sample_texts = [ | |
| # Simple menu item | |
| "Cheeseburger - $10.99\nServed with fries and a pickle.", | |
| # Menu section | |
| "APPETIZERS\n-----------\nMozzarella Sticks - $7.99\nLoaded Nachos - $9.99\nBuffalo Wings - $12.99", | |
| # Complex menu with formatting | |
| """MAIN COURSE | |
| ------------- | |
| Grilled Salmon - $18.99 | |
| Fresh Atlantic salmon served with seasonal vegetables and rice pilaf. | |
| Filet Mignon - $24.99 | |
| 8oz center-cut filet served with mashed potatoes and asparagus. | |
| Vegetable Pasta - $14.99 | |
| Penne pasta with seasonal vegetables in a creamy garlic sauce.""" | |
| ] | |
| results = [] | |
| for i, text in enumerate(sample_texts): | |
| print(f"\nTesting sample {i+1}...") | |
| result = test_braille_translation(text) | |
| results.append(result) | |
| # Print progress | |
| status = "SUCCESS" if result['success'] else "FAILED" | |
| print(f"Sample {i+1}: {status}") | |
| print(f"Words: {result['word_count']}, Time: {result['processing_time']:.2f}s") | |
| print(f"Braille sample: {result['braille_text'][:50]}...") | |
| return results | |
| if __name__ == "__main__": | |
| print("Testing Braille translation functionality...") | |
| results = run_braille_tests() | |
| # Print summary | |
| success_count = sum(1 for r in results if r['success']) | |
| print(f"\nSummary: {success_count}/{len(results)} tests passed") | |
| if results: | |
| avg_time = sum(r['processing_time'] for r in results) / len(results) | |
| print(f"Average processing time: {avg_time:.2f} seconds") | |