Update app.py

app.py

@@ -495,6 +495,24 @@ async def handle_user_input(websocket,key, process, connected,process_ids):
                             response = {"error": "Unsupported method"}
                         # Parse HTML content
                         soup = BeautifulSoup(response.text, "html.parser")
+                        # Function to join relative URLs with base URL
+                        def join_url_with_base(url):
+                            if url.startswith('/'):
+                                return urljoin(url, base_host)
+                            else:
+                                return urljoin(url, base_url.geturl())
+                        
+                        # Find and modify CSS links
+                        for link in soup.find_all("link", rel="stylesheet", href=True):
+                            link['href'] = join_url_with_base(link['href'])
+                        
+                        # Find and modify JS links
+                        for script in soup.find_all("script", src=True):
+                            script['src'] = join_url_with_base(script['src'])
+                        
+                        # Find and modify regular anchor links
+                        for anchor in soup.find_all("a", href=True):
+                            anchor['href'] = join_url_with_base(anchor['href'])
                         # Find and modify CSS links
                         for link in soup.find_all("link", rel="stylesheet", href=True):
                             if link['href'].startswith('/'):
@@ -864,6 +882,24 @@ async def exe(websocket,connected,key,process_ids):
                     response = {"error": "Unsupported method"}
                 # Parse HTML content
                 soup = BeautifulSoup(response.text, "html.parser")
+                # Function to join relative URLs with base URL
+                def join_url_with_base(url):
+                    if url.startswith('/'):
+                        return urljoin(url, base_host)
+                    else:
+                        return urljoin(url, base_url.geturl())
+                
+                # Find and modify CSS links
+                for link in soup.find_all("link", rel="stylesheet", href=True):
+                    link['href'] = join_url_with_base(link['href'])
+                
+                # Find and modify JS links
+                for script in soup.find_all("script", src=True):
+                    script['src'] = join_url_with_base(script['src'])
+                
+                # Find and modify regular anchor links
+                for anchor in soup.find_all("a", href=True):
+                    anchor['href'] = join_url_with_base(anchor['href'])
                 # Find and modify CSS links
                 for link in soup.find_all("link", rel="stylesheet", href=True):
                     if link['href'].startswith('/'):