mirror of
https://github.com/harry0703/MoneyPrinterTurbo.git
synced 2026-02-22 00:47:22 +08:00
Here's a summary of the changes:
1. **Web UI (`webui/Main.py`):**
* I've added a new "Content Generation" section in the middle panel of your application.
* This section includes a text input for your prompts and a button to start the generation process.
* The generated content will be displayed in a text area.
* I've used session state to manage the prompt input and the generated output.
* Localization has been integrated using the `tr()` function.
2. **LLM Service (`app/services/llm.py`):**
* I introduced a new function `generate_content(prompt: str) -> str`.
* This function takes your prompt, formats it for the LLM, and then uses the existing `_generate_response` helper to communicate with the LLM provider.
* I've included logging for prompt processing and to track successful or failed operations.
* Error handling for interactions with the LLM layer has been implemented.
3. **Unit Tests (`test/services/test_llm.py`):**
* I created a new test file specifically for the LLM service.
* I've added unit tests for the `generate_content` function, ensuring it handles:
* Successful content generation.
* Failures resulting from LLM errors.
* The scenario of empty prompts.
* These tests mock the `_generate_response` function to focus on the `generate_content` logic.
* All tests are currently passing.
This new feature offers you another way to utilize the application's LLM capabilities, expanding beyond video script generation.
|
||
|---|---|---|
| .. | ||
| resources | ||
| services | ||
| __init__.py | ||
| README.md | ||
MoneyPrinterTurbo Test Directory
This directory contains unit tests for the MoneyPrinterTurbo project.
Directory Structure
services/: Tests for components in theapp/servicesdirectorytest_video.py: Tests for the video servicetest_task.py: Tests for the task servicetest_voice.py: Tests for the voice service
Running Tests
You can run the tests using Python’s built-in unittest framework:
# Run all tests
python -m unittest discover -s test
# Run a specific test file
python -m unittest test/services/test_video.py
# Run a specific test class
python -m unittest test.services.test_video.TestVideoService
# Run a specific test method
python -m unittest test.services.test_video.TestVideoService.test_preprocess_video
Adding New Tests
To add tests for other components, follow these guidelines:
- Create test files prefixed with
test_in the appropriate subdirectory - Use
unittest.TestCaseas the base class for your test classes - Name test methods with the
test_prefix
Test Resources
Place any resource files required for testing in the test/resources directory.