pdfai / app.py
gpcerv's picture
Update app.py
3fcafd8
raw
history blame contribute delete
No virus
2.07 kB
from flask import Flask
from flask import request
from flask import render_template
from flask import redirect
from flask import jsonify
from flask import send_from_directory
import json
import numpy as np
from numpy.linalg import norm
import datetime
import tensorflow as tf
import tensorflow_hub as hub
import tensorflow_text
from functools import wraps
import os
import pickle
import openai
import requests, zipfile, io
model_url = "./hub_model/"
hub_layer = hub.load(model_url)
word=np.load('./static/encodedT.npy',allow_pickle=True)
with open ('./static/textT', 'rb') as fp:
text = pickle.load(fp)
#word_orig=np.array(word, dtype="object")
app = Flask(__name__)
link=os.environ.get('link')
#print('il link è:',link)
r = requests.get(link)
z = zipfile.ZipFile(io.BytesIO(r.content))
z.extractall("moduli")
from moduli import pdfricerca
def support_jsonp(func):
"""Wraps JSONified output for JSONP requests."""
@wraps(func)
def decorated_function(*args, **kwargs):
callback = request.args.get('callback', False)
if callback:
resp = func(*args, **kwargs)
resp.set_data('{}({})'.format(
str(callback),
resp.get_data(as_text=True)
))
resp.mimetype = 'application/javascript'
return resp
else:
return func(*args, **kwargs)
return decorated_function
@app.route("/")
def flask_app():
return '<br>' \
'<h3><a href="https://www.aubaypoc.altervista.org/chatpdf/" target="_blank">Hai acceso il server! Bene!&#128521 Adesso vai alla pagina CHAT whith my PDF!!</a></h3>' \
@app.route('/requestsSearch',methods=['POST','GET'])
@support_jsonp
def tasksSearch():
domanda=request.args.get('frase').strip()
print('domanda',domanda)
dati=[]
w,out=pdfricerca.ricerca(domanda,word,text,hub_layer)
d={}
d['text']=str(w)
d['chi']=out
dati.append(d)
output=dati
response = jsonify(output)
return response
if __name__ == "__main__":
app.run(host='0.0.0.0', port=7860)