Update main.py

main.py

@@ -91,34 +91,10 @@ def proxy(path):
         # Check if response is HTML and filter content if needed
         content_type = resp.headers.get('Content-Type', '')
         if 'text/html' in content_type:
-            # Create response headers
-            response_headers = {}
-            for key, value in resp.headers.items():
-                if key.lower() not in ['content-encoding', 'content-length', 'transfer-encoding']:
-                    response_headers[key] = value
             # Parse HTML content
             html_content = resp.content.decode('utf-8', errors='ignore')
             soup = BeautifulSoup(html_content, 'html.parser')
             
-            # Ensure basic HTML structure exists
-            if not soup.html:
-                new_html = soup.new_tag('html')
-                soup.append(new_html)
-                soup.html.append(soup.new_tag('head'))
-                soup.html.append(soup.new_tag('body'))
-                # Move all top-level content into body
-                for element in list(soup.contents):
-                    if element.name != 'html':
-                        soup.body.append(element)
-            elif not soup.head:
-                soup.html.insert(0, soup.new_tag('head'))
-            elif not soup.body:
-                soup.html.append(soup.new_tag('body'))
-                # Move all content after head into body
-                for element in list(soup.html.contents):
-                    if element.name != 'head' and element != soup.body:
-                        soup.body.append(element)
-            
             # Filter out "Полная версия ETKA"
             for element in soup.find_all(string=re.compile('Полная версия ETKA')):
                 # Replace the text with empty string
@@ -268,21 +244,10 @@ def proxy(path):
             '''
             if soup.body:
                 soup.body.append(script_tag)
-            
+
             # Add modal container for fullscreen image view
             modal_div = soup.new_tag('div')
             modal_div['id'] = 'imageModal'
-            
-            # Convert modified HTML back to string
-            modified_html = str(soup)
-            
-            # Return modified content with preserved headers
-            return Response(
-                modified_html,
-                status=resp.status_code,
-                headers=response_headers,
-                mimetype='text/html'
-            )
             modal_div['class'] = 'modal'
             modal_div['style'] = 'display: none;'