Update ContentGradio.py

ContentGradio.py

@@ -1,8 +1,6 @@
 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}"
@@ -27,74 +25,75 @@ def get_example():
 
     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:")
+class ContentAgentUI:
+    def __init__(self):
+        # Set the path to the external CSS file
+        css_path = os.path.join(os.getcwd(), "ui", "styles.css")
         
-    # 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>")
+        self.ca_gui = gr.Blocks(css=css_path)
+        self.sections = [
+            self.create_header,
+            self.create_user_guidance,
+            self.create_main,
+            self.create_examples,
+            self.create_footer,
+        ]
         
+        for section in self.sections:
+            section()
+        
+        self.launch()
+
+    def create_header(self):
+        agent_header = """
+        #Content Agent                
+        """
+        with self.ca_gui:
+            gr.Markdown("<div id='header'>" + agent_header + "</div>")
+
+    def create_user_guidance(self):
+        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 self.ca_gui:
+            gr.Markdown("<div id='user-guidance'>" + guidance + "</div>")
+
+    def create_main(self):
+        with self.ca_gui:
+            with gr.Row():
+                with gr.Column():
+                    self.user_input = gr.Textbox(label="Your Input", placeholder="Enter something here...")
+                    self.submit_button = gr.Button("Submit")
+                    self.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
+                    self.submit_button.click(process_input, inputs=self.user_input, outputs=self.output)
+                    self.user_input.submit(process_input, inputs=self.user_input, outputs=self.output)
+
+    def create_examples(self):
+        # 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
+        with self.ca_gui:
+            example_radio.change(fn=lambda example: example, inputs=example_radio, outputs=self.user_input)
 
-# 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
+    def create_footer(self):
+        with self.ca_gui:
+            gr.Markdown("<div id='footer'>Thanks for trying it out!</div>")
 
-    # Launch the Gradio interface
-    ca_gui.launch()
+    def launch(self):
+        self.ca_gui.launch()
 
+    def pass_through_agent(self, agent):
+        # Code to pass through the agent
+        pass