Update main.py

main.py

@@ -91,10 +91,34 @@ 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
@@ -248,6 +272,17 @@ def proxy(path):
             # 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;'