import os
import sys

# Add parent directory to path to import shared modules
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))

from visual_qc_apps.flask_app_template import FlaskAppTemplate

class WordCountApp(FlaskAppTemplate):
    """
    Word Count Check
    """
    
    def __init__(self):
        # Define the hardcoded prompt
        prompt = """You are performing a visual quality-control check on a Point of Sale (POS) advertisement. Your task is to determine whether the advertisement contains 7 words or fewer. Examine the advertisement carefully and follow these steps:

STEPS TO EVALUATE:
1. Identify all textual elements in the advertisement, excluding any terms and conditions, the word "NEW," brand logos, and any text present on product packaging.
2. Count the total number of words in these textual elements.
3. Determine if the count is 7 words or fewer.

YOUR OUTPUT:
Based on the evaluation, confirm whether the advertisement passes the word-count checkpoint, stating "Pass" if it contains 7 words or fewer, or "Fail" if it contains more.

Create a JSON code block with these fields:
{ 
  "word_count": (total number of words counted), 
  "checkpoint_result": "Pass" or "Fail" 
}"""
        
        # Initialize the Flask app with the prompt
        super().__init__(__name__, prompt)

# Run the app if executed directly
if __name__ == "__main__":
    app = WordCountApp()
    app.run(debug=True)