gmshroff commited on
Commit
5137230
·
1 Parent(s): e0a860b

made into pure anvil server

Browse files
Files changed (1) hide show
  1. app.py +48 -46
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from flask import Flask,request,render_template,send_file,jsonify
2
  import os
3
  from transformers import AutoTokenizer, AutoModel
4
  import anvil.server
@@ -10,7 +10,7 @@ from background_service import BackgroundTaskService
10
 
11
  anvil.server.connect('PLMOIU5VCGGUOJH2XORIBWV3-ZXZVFLWX7QFIIAF4')
12
 
13
- app=Flask(__name__)
14
  MESSAGED={'title':'API Server',
15
  'messageL':['published server functions:','encode(text)',
16
  'call_gemini(text,key)','call_gpt(text,key,model)',
@@ -72,49 +72,51 @@ def print_results_table():
72
  global service
73
  return(service.results)
74
 
75
- @app.route('/encode',methods=['GET','POST'])
76
- def encode():
77
- print(request)
78
- if request.method=='GET':
79
- text=request.args.get('text')
80
- elif request.method=='POST':
81
- data=request.get_json()
82
- if 'text' in data: text=data["text"]
83
- if text=='' or text is None: return -1
84
- inputs = tokenizer(text, padding=True, truncation=True,
85
- return_tensors="pt", max_length=512)
86
- result = encoder(**inputs)
87
- embeddings = result.last_hidden_state[:, 0, :]
88
- emb_array = embeddings.detach().numpy()
89
- embedding=emb_array.tolist()
90
- return jsonify({'embedding': embedding})
91
-
92
- @app.route("/file/<string:filename>")
93
- def return_file(filename):
94
- return send_file('./data/'+filename)
95
-
96
- @app.route('/run',methods=['GET','POST'])
97
- def run_script():
98
- script=''
99
- # print(request.method)
100
- print(request)
101
- if request.method=='GET':
102
- script=request.args.get('script')
103
- print('I am in get')
104
- elif request.method=='POST':
105
- print('I am in post')
106
- data=request.get_json()
107
- if 'script' in data: script=data['script']
108
- if script=='' or script is None: return 'INVALID'
109
- os.system(script+' > ./out.txt')
110
- with open('./out.txt','r') as f: output=f.read()
111
- return output
112
-
113
- @app.route('/',methods=['GET', 'POST'])
114
- def home():
115
- return render_template('home.html',messageD=MESSAGED)
116
-
117
- if __name__=='__main__':
118
- app.run(host="0.0.0.0", port=7860)
 
 
119
 
120
 
 
1
+ # from flask import Flask,request,render_template,send_file,jsonify
2
  import os
3
  from transformers import AutoTokenizer, AutoModel
4
  import anvil.server
 
10
 
11
  anvil.server.connect('PLMOIU5VCGGUOJH2XORIBWV3-ZXZVFLWX7QFIIAF4')
12
 
13
+ # app=Flask(__name__)
14
  MESSAGED={'title':'API Server',
15
  'messageL':['published server functions:','encode(text)',
16
  'call_gemini(text,key)','call_gpt(text,key,model)',
 
72
  global service
73
  return(service.results)
74
 
75
+ anvil.server.wait_forever()
76
+
77
+ # @app.route('/encode',methods=['GET','POST'])
78
+ # def encode():
79
+ # print(request)
80
+ # if request.method=='GET':
81
+ # text=request.args.get('text')
82
+ # elif request.method=='POST':
83
+ # data=request.get_json()
84
+ # if 'text' in data: text=data["text"]
85
+ # if text=='' or text is None: return -1
86
+ # inputs = tokenizer(text, padding=True, truncation=True,
87
+ # return_tensors="pt", max_length=512)
88
+ # result = encoder(**inputs)
89
+ # embeddings = result.last_hidden_state[:, 0, :]
90
+ # emb_array = embeddings.detach().numpy()
91
+ # embedding=emb_array.tolist()
92
+ # return jsonify({'embedding': embedding})
93
+
94
+ # @app.route("/file/<string:filename>")
95
+ # def return_file(filename):
96
+ # return send_file('./data/'+filename)
97
+
98
+ # @app.route('/run',methods=['GET','POST'])
99
+ # def run_script():
100
+ # script=''
101
+ # # print(request.method)
102
+ # print(request)
103
+ # if request.method=='GET':
104
+ # script=request.args.get('script')
105
+ # print('I am in get')
106
+ # elif request.method=='POST':
107
+ # print('I am in post')
108
+ # data=request.get_json()
109
+ # if 'script' in data: script=data['script']
110
+ # if script=='' or script is None: return 'INVALID'
111
+ # os.system(script+' > ./out.txt')
112
+ # with open('./out.txt','r') as f: output=f.read()
113
+ # return output
114
+
115
+ # @app.route('/',methods=['GET', 'POST'])
116
+ # def home():
117
+ # return render_template('home.html',messageD=MESSAGED)
118
+
119
+ # if __name__=='__main__':
120
+ # app.run(host="0.0.0.0", port=7860)
121
 
122