diff --git a/ayct_backend/__init__.py b/ayct_backend/__init__.py index f12cad3..d96bf53 100644 --- a/ayct_backend/__init__.py +++ b/ayct_backend/__init__.py @@ -22,7 +22,6 @@ def create_app(): campaign_db.init_app(app) twitter_db.create_all(app=app) - campaign_db.drop_all(app=app) campaign_db.create_all(app=app) app.register_blueprint(twitter, url_prefix='/twitter') diff --git a/ayct_backend/campaign/__init__.py b/ayct_backend/campaign/__init__.py index 6c1edd5..23fe447 100644 --- a/ayct_backend/campaign/__init__.py +++ b/ayct_backend/campaign/__init__.py @@ -1,4 +1,5 @@ -import os +import requests +import json from requests_oauthlib import OAuth1Session from flask import Blueprint, current_app, request, jsonify from ayct_backend.twitter.models import * @@ -38,8 +39,17 @@ def add_twitter_account(): twitter_account = TwitterAccount.query.filter_by(twitter_account_id=json['twitter_account_id']).first() - # generate campaign content, change after core integration - generated_content = "This is wonderful post content generated by our very smart core API!" + # generate campaign content + core_url = 'http://65.108.80.28:4999/generate' + payload = { + "data": json['user_input'], + "length": 400 + } + + response = requests.post(core_url, headers={"Content-Type":"application/json"}, data=json.dumps(payload)) + + generated_content = response.content.decode("utf-8").replace('<|endoftext|>', '') + #generated_content = "This is wonderful post content generated by our very smart core API!" # create post on twitter oauth = OAuth1Session( @@ -51,7 +61,7 @@ def add_twitter_account(): response = oauth.post( "https://api.twitter.com/2/tweets", - json={"text": generated_content}, + json={"text": generated_content[0:260]}, ).json()['data'] # save campaign to database