Commit 5493d152 authored by Sergey Pinus's avatar Sergey Pinus
Browse files

update proxy

parent 2d72206a

modules/ocr/ocr_google_lens.py

+2 −11
Original line number Diff line number Diff line
@@ -5,7 +5,6 @@ import cv2 
from typing import List



import httpx

from httpx_socks import SyncProxyTransport

from PIL import Image

import io

import json5
@@ -35,16 +34,8 @@ class LensCore: 


    def _send_request(self, url, headers, files):

        try:

            if self.proxy:

                if self.proxy.startswith('socks5://') or self.proxy.startswith('socks4://'):

                    transport = SyncProxyTransport.from_url(self.proxy)

                    client = httpx.Client(transport=transport)

            client = httpx.Client(proxies=self.proxy) if self.proxy else httpx.Client()

            response = client.post(url, headers=headers, files=files)

                else:

                    response = httpx.post(url, headers=headers, files=files, proxies=self.proxy)

            else:

                response = httpx.post(url, headers=headers, files=files)



            if response.status_code != 200:

                raise Exception(f"Failed to upload image. Status code: {response.status_code}")

            return response