Update app.py

app.py

@@ -9,32 +9,32 @@ WEBSITES = {
 def open_website(website_name):
     """
     Function to retrieve the website URL from the dictionary.
-    Handles cases where the website name doesn't exist.
+    It embeds the selected website URL in an HTML iframe for interactive viewing.
     """
     url = WEBSITES.get(website_name)
     if url:
-        return url
+        # Embed the website URL inside an iframe for interactive viewing
+        return f'<iframe src="{url}" width="100%" height="600"></iframe>'
     else:
-        return "Website not found."
+        return "<p>Website not found.</p>"
 
 # Define the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Website Selector Tool")
     gr.Markdown("""
-    This tool allows you to select a website and view its URL.
-    Click the 'Open Website' button to get the link!
+    This tool allows you to select and interact with a website in an embedded viewer.
     """)
     
-    # Dropdown for selecting website
+    # Dropdown for selecting a website
     website_name = gr.Dropdown(list(WEBSITES.keys()), label="Select a Website")
     
-    # Button to trigger website retrieval
+    # Button to trigger website embedding
     open_btn = gr.Button("Open Website")
     
-    # Textbox to display the URL of the selected website
-    result = gr.Textbox(label="Website URL", interactive=False)
+    # HTML component to display the selected website in an iframe
+    result = gr.HTML(label="Interactive Website Viewer")
     
-    # Click event that retrieves the website URL and displays it
+    # Click event to retrieve the website and embed it in the iframe
     open_btn.click(open_website, inputs=website_name, outputs=result)
 
 demo.launch()