File size: 1,833 Bytes
dbd2ac6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import os
import sys
import time
import webbrowser

print('__file__: %s' % __file__)
path1 = os.path.dirname(os.path.abspath(__file__))
sys.path.append(path1)
base_path = os.path.dirname(path1)
sys.path.append(base_path)
os.environ['PYTHONPATH'] = path1
print('path1', path1, flush=True)

os.environ['NLTK_DATA'] = os.path.join(base_path, './nltk_data')
os.environ['PATH'] = os.environ['PATH'] + ';' + \
                     os.path.join(base_path, 'poppler/Library/bin/') + ';' + \
                     os.path.join(base_path, 'poppler/Library/lib/') + ';' + \
                     os.path.join(base_path, 'Tesseract-OCR') + \
                     os.path.join(base_path, 'ms-playwright') + \
                     os.path.join(base_path, 'ms-playwright/chromium-1076/chrome-win') + \
                     os.path.join(base_path, 'ms-playwright/ffmpeg-1009') + \
                     os.path.join(base_path, 'ms-playwright/firefox-1422/firefox') + \
                     os.path.join(base_path, 'ms-playwright/webkit-1883')
print(os.environ['PATH'])

for sub in ['src', 'iterators', 'gradio_utils', 'metrics', 'models', '.']:
    path2 = os.path.join(base_path, '..', sub)
    sys.path.append(path2)
    print(path2, flush=True)

    path2 = os.path.join(path1, '..', sub)
    sys.path.append(path2)
    print(path2, flush=True)


def main():
    from generate import entrypoint_main as main_h2ogpt
    os.environ['h2ogpt_block_gradio_exit'] = 'False'
    os.environ['h2ogpt_score_model'] = ''
    main_h2ogpt()

    server_name = os.getenv('h2ogpt_server_name', os.getenv('H2OGPT_SERVER_NAME', 'localhost'))
    server_port = os.getenv('GRADIO_SERVER_PORT', str(7860))

    url = "http://%s:%s" % (server_name, server_port)
    webbrowser.open(url)

    while True:
        time.sleep(10000)


if __name__ == "__main__":
    main()