import gradio as gr
import os



# Function for Main content (takes user input and returns a response)
def process_input(user_input):
    return f"You entered: {user_input}"

# Function to generate predefined examples
def get_example():
    # Define the path to the 'examples' directory
    example_root = os.path.join(os.path.dirname(__file__), "examples")
    
    # Get list of all example text file paths
    example_files = [os.path.join(example_root, _) for _ in os.listdir(example_root) if _.endswith("txt")]
    
    # Read the content of each file (assuming they're plain text files)
    examples = []
    
    for file_path in example_files:
        example_content = ""
        with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
            example_content = f.read()
        
        examples.append(example_content)  # Append the content to the list

    return examples

# Create the header section
def create_header():

    agent_header = """
    #Content Agent                
    """
    with gr.Row():
        gr.Markdown("<div id='header'>" + agent_header + "</div>")

# Create the user guidance section
def create_user_guidance():

    guidance = """
    Please enter text below to get started. The AI Agent will try to determine whether the language is polite and uses the following classification:
    - `polite`
    - `somewhat polite`
    - `neutral`
    - `impolite`

    App is running `deepseek-ai/DeepSeek-R1-Distill-Qwen-32B` text generation model.
    Uses Intel's Polite Guard NLP library.
    Compute is GCP · Nvidia L4 · 4x GPUs · 96 GB
    """
    with gr.Row():
        gr.Markdown("<div id='user-guidance'>" + guidance+ "</div>")

# Create the main content section
def create_main():
    with gr.Row():
        with gr.Column():
            user_input = gr.Textbox(label="Your Input", placeholder="Enter something here...")
            submit_button = gr.Button("Submit")
            output = gr.Textbox(label="Content feedback", interactive=False, lines=10, max_lines=20 )

            # Define the function to be called when the button is clicked or Enter is pressed
            submit_button.click(process_input, inputs=user_input, outputs=output)
            user_input.submit(process_input, inputs=user_input, outputs=output)
            
    return user_input, output  # Return both input and output components

# Create the examples section
def create_examples(user_input):
    # Fetch examples by calling get_example() here
    examples = get_example()
    example_radio = gr.Radio(choices=examples, label="Try one of these examples:")
        
    # When an example is selected, populate the input field
    example_radio.change(fn=lambda example: example, inputs=example_radio, outputs=user_input)

# Create the footer section
def create_footer():
    with gr.Row():
        gr.Markdown("<div id='footer'>Thanks for trying it out!</div>")
        

# Main function to bring all sections together
def ContentAgentUI():
    # Set the path to the external CSS file
    css_path = os.path.join(os.getcwd(), "ui", "styles.css")
    
    with gr.Blocks(css=css_path) as ca_gui:
        create_header()  # Create the header
        create_user_guidance()  # Create user guidance section
        user_input, output = create_main()  # Create the main section (returns the input/output components)
        create_examples(user_input)  # Create the examples section
        create_footer()  # Create the footer section

    # Launch the Gradio interface
    ca_gui.launch()