Update app.py

app.py

@@ -136,7 +136,7 @@ def ip_swapper_handler(original_url, single_ip, multiple_ips, test_connectivity)
     """Main handler function for IP swapping"""
     try:
         if not original_url.strip():
-            return "Please enter a URL", "", "", gr.update(visible=False)
+            return "Please enter a URL", "", "", gr.update(visible=False), "about:blank", "No URL to display"
         
         # Extract original IP for display
         original_ip = extract_ip_from_url(original_url)
@@ -146,18 +146,18 @@ def ip_swapper_handler(original_url, single_ip, multiple_ips, test_connectivity)
         
         if single_ip.strip():
             if not is_valid_ip(single_ip):
-                return f"Error: Invalid IP address format: {single_ip}", "", "", gr.update(visible=False)
+                return f"Error: Invalid IP address format: {single_ip}", "", "", gr.update(visible=False), "about:blank", "Invalid IP format"
             ips_to_process.append(single_ip)
         
         if multiple_ips.strip():
             ip_list = [ip.strip() for ip in multiple_ips.split('\n') if ip.strip()]
             for ip in ip_list:
                 if not is_valid_ip(ip):
-                    return f"Error: Invalid IP address format: {ip}", "", "", gr.update(visible=False)
+                    return f"Error: Invalid IP address format: {ip}", "", "", gr.update(visible=False), "about:blank", f"Invalid IP: {ip}"
             ips_to_process.extend(ip_list)
         
         if not ips_to_process:
-            return "Please enter at least one valid IP address", "", "", gr.update(visible=False)
+            return "Please enter at least one valid IP address", "", "", gr.update(visible=False), "about:blank", "No IPs provided"
         
         # Process URLs
         results = []
@@ -169,6 +169,8 @@ def ip_swapper_handler(original_url, single_ip, multiple_ips, test_connectivity)
                 'original_url': original_url
             })
             result_text = new_url
+            iframe_url = new_url
+            iframe_title = f"Testing: {ips_to_process[0]}"
         else:
             result_lines = []
             for i, ip in enumerate(ips_to_process, 1):
@@ -182,6 +184,9 @@ def ip_swapper_handler(original_url, single_ip, multiple_ips, test_connectivity)
                 result_lines.append(new_url)
                 result_lines.append("")
             result_text = "\n".join(result_lines)
+            # For multiple IPs, show the first one in iframe
+            iframe_url = results[0]['url'] if results else "about:blank"
+            iframe_title = f"Testing: {results[0]['ip']}" if results else "Multiple IPs"
         
         # Get decoded r parameter
         decoded_r = decode_r_parameter(original_url)
@@ -243,10 +248,10 @@ def ip_swapper_handler(original_url, single_ip, multiple_ips, test_connectivity)
         # Show history button if we have test results
         history_visible = len(test_results) > 0
         
-        return original_url, result_text, decoded_r, gr.update(visible=history_visible)
+        return original_url, result_text, decoded_r, gr.update(visible=history_visible), iframe_url, iframe_title
         
     except Exception as e:
-        return f"Error: {str(e)}", "", "", gr.update(visible=False)
+        return f"Error: {str(e)}", "", "", gr.update(visible=False), "about:blank", f"Error: {str(e)}"
 
 def get_history_df():
     """Convert history data to DataFrame for display"""
@@ -268,7 +273,7 @@ def get_history_df():
 
 def clear_all():
     """Clear all inputs and outputs"""
-    return "", "", "", "", "", gr.update(visible=False)
+    return "", "", "", "", "", gr.update(visible=False), "about:blank", "Ready for testing..."
 
 def show_history():
     """Display history in a DataFrame"""
@@ -298,6 +303,21 @@ def export_history():
     df.to_csv(filename, index=False)
     return filename
 
+def load_url_in_iframe(url, ip_address):
+    """Load a specific URL in the iframe"""
+    if not url or url.startswith("Error:") or url.startswith("Please enter"):
+        return "about:blank", "No valid URL to display"
+    
+    # Validate URL format
+    try:
+        parsed = urlparse(url)
+        if parsed.scheme and parsed.netloc:
+            return url, f"Testing: {ip_address}"
+        else:
+            return "about:blank", "Invalid URL format"
+    except:
+        return "about:blank", "Invalid URL"
+
 # Custom CSS for better styling
 custom_css = """
 .gradio-container {
@@ -349,6 +369,39 @@ custom_css = """
 .test-btn {
     background: linear-gradient(90deg, #10b981, #059669) !important;
 }
+.iframe-container {
+    border: 2px solid #e5e7eb;
+    border-radius: 10px;
+    overflow: hidden;
+    margin-top: 10px;
+}
+.iframe-header {
+    background: #4f46e5;
+    color: white;
+    padding: 8px 12px;
+    font-weight: bold;
+    display: flex;
+    justify-content: between;
+    align-items: center;
+}
+.iframe-content {
+    height: 500px;
+    width: 100%;
+}
+.url-test-buttons {
+    margin: 10px 0;
+    display: flex;
+    gap: 10px;
+    flex-wrap: wrap;
+}
+.url-test-btn {
+    background: #4f46e5 !important;
+    color: white !important;
+    border: none !important;
+}
+.url-test-btn:hover {
+    background: #4338ca !important;
+}
 """
 
 # Create Gradio interface
@@ -360,7 +413,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
             🔄 IP Swapper Wizard
         </h1>
         <p style="color: #6B7280; font-size: 1.1rem;">
-            Swap IP addresses in URLs while maintaining all parameters • Test Connectivity • Track History
+            Swap IP addresses in URLs while maintaining all parameters • Test Connectivity • Track History • Live Preview
         </p>
     </div>
     """)
@@ -432,6 +485,38 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                     variant="secondary"
                 )
     
+    # Live Preview Section
+    with gr.Accordion("🌐 Live URL Preview & Testing", open=True):
+        gr.Markdown("**Test your generated URLs directly in the browser below:**")
+        
+        with gr.Row():
+            with gr.Column(scale=3):
+                iframe_title = gr.Textbox(
+                    label="Current Preview:",
+                    interactive=False,
+                    max_lines=1,
+                    value="Ready for testing..."
+                )
+                
+                # URL testing buttons will be dynamically generated
+                url_test_buttons = gr.HTML(
+                    value="<div style='padding: 10px; background: #f3f4f6; border-radius: 5px; text-align: center;'>Generated URLs will appear here after processing</div>"
+                )
+            
+            with gr.Column(scale=7):
+                iframe_component = gr.HTML(
+                    value="""
+                    <div class="iframe-container">
+                        <div class="iframe-header">
+                            <span>🔍 Live Browser Preview</span>
+                        </div>
+                        <iframe src="about:blank" class="iframe-content" id="preview-iframe" 
+                                sandbox="allow-scripts allow-same-origin allow-forms allow-popups"
+                                loading="lazy"></iframe>
+                    </div>
+                    """
+                )
+    
     # History Section
     with gr.Accordion("📊 Test History & Analytics", open=False):
         with gr.Row():
@@ -450,14 +535,14 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
         3. **Check 'Test connectivity'** to automatically test URLs
         4. **Click 'Swap IP(s)'** to process and test
         5. **View results** with status indicators (✅ Working / ❌ Not Working)
-        6. **Check history** for past test results
+        6. **Test URLs live** in the built-in browser
+        7. **Check history** for past test results
 
-        **Testing Features:**
-        - Automatic HTTP connectivity testing
-        - Response time measurement
-        - Status code reporting
-        - Historical tracking with success rates
-        - Export results to CSV
+        **Live Preview Features:**
+        - Test URLs directly in the app
+        - Built-in browser with iframe
+        - Quick navigation between multiple IPs
+        - Safe sandboxed environment
 
         **Status Indicators:**
         - ✅ Working: URL responded with HTTP 200
@@ -468,19 +553,35 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     process_btn.click(
         fn=ip_swapper_handler,
         inputs=[url_input, single_ip_input, multiple_ips_input, test_connectivity],
-        outputs=[original_url_output, modified_url_output, decoded_r_output, history_btn]
+        outputs=[original_url_output, modified_url_output, decoded_r_output, history_btn, iframe_component, iframe_title]
+    ).then(
+        fn=lambda modified_url, single_ip, multiple_ips: generate_url_buttons(modified_url, single_ip, multiple_ips),
+        inputs=[modified_url_output, single_ip_input, multiple_ips_input],
+        outputs=[url_test_buttons]
     )
     
     decode_btn.click(
-        fn=lambda url: (url, "", decode_r_parameter(url), gr.update(visible=False)) if url else ("", "", "Please enter a URL", gr.update(visible=False)),
+        fn=lambda url: (url, "", decode_r_parameter(url), gr.update(visible=False), "about:blank", "Decode Only Mode") if url else ("", "", "Please enter a URL", gr.update(visible=False), "about:blank", "No URL"),
         inputs=[url_input],
-        outputs=[original_url_output, modified_url_output, decoded_r_output, history_btn]
+        outputs=[original_url_output, modified_url_output, decoded_r_output, history_btn, iframe_component, iframe_title]
+    ).then(
+        fn=lambda modified_url, single_ip, multiple_ips: "<div style='padding: 10px; background: #f3f4f6; border-radius: 5px; text-align: center;'>Use 'Swap IP(s)' to generate testable URLs</div>",
+        inputs=[modified_url_output, single_ip_input, multiple_ips_input],
+        outputs=[url_test_buttons]
     )
     
     clear_btn.click(
         fn=clear_all,
         inputs=[],
-        outputs=[url_input, single_ip_input, multiple_ips_input, original_url_output, modified_url_output, decoded_r_output, history_btn]
+        outputs=[url_input, single_ip_input, multiple_ips_input, original_url_output, modified_url_output, decoded_r_output, history_btn, iframe_title]
+    ).then(
+        fn=lambda: "about:blank",
+        inputs=[],
+        outputs=[iframe_component]
+    ).then(
+        fn=lambda: "<div style='padding: 10px; background: #f3f4f6; border-radius: 5px; text-align: center;'>Generated URLs will appear here after processing</div>",
+        inputs=[],
+        outputs=[url_test_buttons]
     )
     
     # History handlers
@@ -508,6 +609,83 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
         outputs=[history_output]
     )
 
+def generate_url_buttons(modified_url, single_ip, multiple_ips):
+    """Generate clickable buttons for testing URLs in iframe"""
+    if not modified_url or modified_url.startswith("Error:") or modified_url.startswith("Please enter"):
+        return "<div style='padding: 10px; background: #f3f4f6; border-radius: 5px; text-align: center;'>No valid URLs to test</div>"
+    
+    # Extract URLs from the modified_url text
+    urls = []
+    ips = []
+    
+    if single_ip and is_valid_ip(single_ip):
+        # Single IP case - extract the URL directly
+        urls.append(modified_url.split('\n')[0])  # Take first line as URL
+        ips.append(single_ip)
+    
+    if multiple_ips:
+        # Multiple IPs case - parse the formatted output
+        ip_list = [ip.strip() for ip in multiple_ips.split('\n') if ip.strip() and is_valid_ip(ip)]
+        lines = modified_url.split('\n')
+        
+        for i, line in enumerate(lines):
+            if line.startswith('http'):
+                # Find the corresponding IP
+                for ip in ip_list:
+                    if ip in lines[i-1] if i > 0 else False:
+                        urls.append(line)
+                        ips.append(ip)
+                        break
+    
+    # If we couldn't parse structured output, try to extract URLs directly
+    if not urls:
+        import re
+        url_pattern = r'https?://[^\s]+'
+        found_urls = re.findall(url_pattern, modified_url)
+        urls = found_urls
+        # Use the provided IPs or generate placeholders
+        if single_ip:
+            ips = [single_ip] * len(urls)
+        elif multiple_ips:
+            ip_list = [ip.strip() for ip in multiple_ips.split('\n') if ip.strip()]
+            ips = ip_list[:len(urls)]
+        else:
+            ips = [f"IP_{i+1}" for i in range(len(urls))]
+    
+    if not urls:
+        return "<div style='padding: 10px; background: #f3f4f6; border-radius: 5px; text-align: center;'>No valid URLs found in output</div>"
+    
+    # Generate buttons HTML
+    buttons_html = '<div class="url-test-buttons">'
+    buttons_html += '<strong>Test URLs:</strong><br>'
+    
+    for i, (url, ip) in enumerate(zip(urls, ips)):
+        button_id = f"url-btn-{i}"
+        buttons_html += f'''
+        <button class="url-test-btn" onclick="loadIframe('{url}', '{ip}')" style="margin: 2px; padding: 5px 10px; border-radius: 4px;">
+            🌐 Test {ip}
+        </button>
+        '''
+    
+    buttons_html += '</div>'
+    buttons_html += '''
+    <script>
+    function loadIframe(url, ip) {
+        const iframe = document.getElementById('preview-iframe');
+        const title = document.querySelector('[data-testid="iframe_title"] input');
+        iframe.src = url;
+        if (title) title.value = 'Testing: ' + ip;
+        
+        // Update Gradio state (optional)
+        if (window.gradio_app) {
+            // You can add Gradio state updates here if needed
+        }
+    }
+    </script>
+    '''
+    
+    return buttons_html
+
 # Launch configuration
 if __name__ == "__main__":
     # Disable SSL warnings for testing