cryptobot-v2 / app.py
alvaromlua's picture
sin st
50554dd verified
from transformers import pipeline
import requests
from flask import Flask, jsonify, request
app = Flask(__name__)
@app.route('/compararnoticias', methods=['POST'])
def get_crypto_news_compare():
fulfillment_text = ""
pipe = pipeline('sentiment-analysis')
# Realizar la solicitud a la API
req = request.get_json()
query_result = req.get('queryResult', {})
parameters = query_result.get('parameters', {})
noticias = req.get('sessionInfo', {}).get('parameters', {}).get('cryptoparam', '').lower()
# Configuración de los parámetros para la solicitud a la API
params = {
'categories': noticias,
}
out = pipe(noticias)
resultado = out[0]['label']
accuracy = out[0]['score']
if resultado == "POSITIVE":
fulfillment_text = f"Mi análisis al {accuracy*100:.2.f}% indica que es buena noticia para esta criptomoneda, puede ser una oportunidad valiosa para vender\n"
else:
fulfillment_text = f"Mi análisis al {accuracy*100:.2.f}% indica que es mala noticia para esta criptomoneda, puede ser una oportunidad para comprar\n"
return jsonify({
"fulfillmentResponse": {
"messages": [{
"text": {
"text": [fulfillment_text]
}
}]
}
})
if __name__ == '__main__':
app.run(port=5000, debug=True)