Spaces:
Runtime error
Runtime error
from flask import Flask, request | |
from generators.title_to_abstract import title_to_abstract_generator | |
from generators.topic_to_abstract import topic_to_abstract_generator | |
app = Flask(__name__) | |
max_length_char = 360 | |
def rewriter(): | |
data = request.get_json(silent=True) | |
if data is None: | |
return 'No data received' | |
else: | |
if data.get('title') is None and data.get('topic') is not None: | |
return generate(data, 'topic') | |
elif data.get('title') is not None: | |
return generate(data, 'title') | |
else: | |
return 'No data received' | |
def generate(data, value): | |
obj = abstract_generator_obj(value) | |
result = obj(template_generator(value, data[value])) | |
if len(result['result']) > 40: | |
return {"result": result['result']} | |
else: | |
result_re = obj(template_generator(value, data[value])) | |
while len(result_re['result']) < 50: | |
print(f"getting results again, {result_re}") | |
result_re = obj(template_generator(value, data[value])) | |
if len(result_re['result']) > 40: | |
return {"result": result_re['result']} | |
def abstract_generator_obj(value): | |
if value == 'title': | |
return title_to_abstract_generator | |
else: | |
return topic_to_abstract_generator | |
def template_generator(type, value): | |
if type == 'title': | |
return "title: " + value + "\n" + 'abstract: ' | |
else: | |
keyword_str = "\n" | |
for k in value: | |
keyword_str = keyword_str + k + "\n" | |
return "topic: " + keyword_str + "\n" + 'abstract: ' | |
if __name__ == '__main__': | |
app.run() | |