Update test_prompt_benchmark_test.py

[Yash-2707]

Error Handling: Adding try-except blocks to handle potential errors.
Logging: Implementing a logging mechanism to track the progress and any important events or errors.
Configurable Model Selection: Allowing the user to specify which models to test dynamically, rather than hardcoding the model list.
Performance Metrics: Collecting and printing performance metrics for the entire benchmark test.
Visualization: Adding a simple visualization component for summarizing results (optional if relevant libraries are available).
Extensibility: Making the code easily extensible for future enhancements.