Try fix by Amazon Q

app.py

@@ -1,39 +1,62 @@
-import gradio as gr 
+import gradio as gr
 
-class State: 
-    def __init__(self): 
-        self.context = "This Adrega component is designed to answer questions provided by API" 
-        self.question = "What does this component do?" 
+class State:
+    def __init__(self):
+        self.context = "This Adrega component is designed to answer questions provided by API"
+        self.question = "What does this component do?"
 
-state = State() 
+state = State()
 
-js_code = """ 
-window.addEventListener("message", on_message); 
-"""
-
-def update_values(context, question): 
+def update_values(context, question):
     state.context = context
     state.question = question
     return "I have stored the new values"
-    
-with gr.Blocks(js = js_code) as demo:
-    iface = gr.Interface( 
-        fn=update_values, 
-        inputs=["text", "text"], 
-        outputs="text"
+
+def process_message(message):
+    # Handle incoming messages
+    state.context = message
+    return message
+
+# JavaScript for message handling
+js = """
+function on_message(event) {
+    if (event.data && event.data.projectData) {
+        // Access the gradio elements through window.gradio
+        const projectData = event.data.projectData;
+        const question = event.data.question;
+        
+        // Update the textboxes
+        document.querySelector('#project_data textarea').value = projectData;
+        document.querySelector('#output textarea').value = question;
+    }
+}
+window.addEventListener('message', on_message);
+"""
+
+with gr.Blocks(js=js) as demo:
+    with gr.Row():
+        project_data = gr.Textbox(
+            label="Project Data",
+            lines=2,
+            value=state.context,
+            elem_id="project_data"
+        )
+        output = gr.Textbox(
+            label="Output",
+            elem_id="output"
+        )
+
+    # Add interface for update_values function
+    update_btn = gr.Button("Update Values")
+    update_btn.click(
+        fn=update_values,
+        inputs=[project_data, output],
+        outputs=output
     )
-    
-    project_data = gr.Textbox(label="Project Data", lines=2, value=state.context) 
-    output = gr.Textbox(label="Output")
-
-    def on_message(event): 
-        # Update the textbox value based on the received message 
-        state.context = event.data 
-        iface.set_value(project_data, value=state.context) 
-        iface.set_value(output, value=state.context)
-
-iface.launch( 
-    server_name="0.0.0.0", # Allow external connections 
-    share=True, # Create public URL 
-    allowed_paths=["*"], #Allow CORS
+
+# Launch the app
+demo.launch(
+    server_name="0.0.0.0",
+    share=True,
+    allowed_paths=["*"]  # Be careful with this in production
 )