Update alternative-2.py

alternative-2.py

@@ -97,7 +97,7 @@ def extractDataFromUrls(urls: str, loader_type: str):
         for item in data:
             jsonData.append(item.to_json())
 
-        return jsonData, data
+        return jsonData, data, urls[0]
 
     except Exception as err:
         return "An Error Occurred. Contact Developer" + str(err), "Error Occured. Boom" 
@@ -140,27 +140,26 @@ choices = [
 
 # Create the Gradio interface
 with gr.Blocks(theme="sudeepshouche/minimalist") as demo:
+    extracted_url = gr.State()  # Use gr.State() to store the URL
+    
     gr.Markdown("# The Big Scraper")
     with gr.Tab("Scraped"):
         with gr.Row():
-            url_input = gr.Textbox(label="Enter your comma separated URLs here")
-            loader_dropdown = gr.Dropdown(choices=choices, label="Pick your Loader from here")
-        with gr.Row():
-            json_output = gr.JSON(label="Extracted Data (JSON)")
-            text_output = gr.Textbox(label="Extracted Data (Text)")
+            with gr.Column():
+                url_input = gr.Textbox(label="Enter your comma separated URLs here")
+                loader_dropdown = gr.Dropdown(choices=choices, label="Pick your Loader from here")
+            with gr.Column():
+                json_output = gr.JSON(label="Extracted Data (JSON)")
+                text_output = gr.Textbox(label="Extracted Data (Text)")
         
         btn = gr.Button("Extract Data")
-        btn.click(extractDataFromUrls, inputs=[url_input, loader_dropdown], outputs=[json_output, text_output])
+        btn.click(extractDataFromUrls, inputs=[url_input, loader_dropdown], outputs=[json_output, text_output, extracted_url]) \
+           .then(take_webdata, inputs=extracted_url, outputs=[screenshot_output, title_output])
 
     with gr.Tab("Images"):
-        with gr.Row():
-            image_url_input = gr.Textbox(label="Enter URL for Screenshot:")
         with gr.Row():
             screenshot_output = gr.Image(label="Screenshot")
             title_output = gr.Textbox(label="Page Title")
         
-        btn2 = gr.Button("Take Screenshot")
-        btn2.click(take_webdata, inputs=image_url_input, outputs=[screenshot_output, title_output])
-
 # Launch the Gradio interface
 demo.launch()