ContentGradio.py

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
 
                Use content agent to determine whether language is polite by passing it text strings. 
            
                Content Agent checks text and classifies it as polite, somewhat polite, neutral, and impolite.  
                Uses Intel's Polite Guard NLP library 
                
                 
                """
    with gr.Row():
        gr.Markdown("<div id='header'>" + agent_header" </div>")

# Create the user guidance section
def create_user_guidance():
    with gr.Row():
        gr.Markdown("<div id='user-guidance'>Please enter  text below to get started. Select one of the examples to see the classfication results.</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...")
            output = gr.Textbox(label="Output", interactive=False)
            # move this down to ensure that output it defined
            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()

    gr.Markdown("<div id='examples'>Try one of these examples:</div>")
    # Create a Radio component with the list of examples
    example_radio = gr.Radio(choices=examples, label="Select an Example")
        
    # 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()