Codegeist/app.py at main · lazyCodes7/Codegeist · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
from graphs.atlassian.GraphRenderer import GraphRenderer
from scraper.atlassian.ReviewScraper import ReviewScraper
from flask import Flask, render_template, send_file, make_response, url_for, Response,request,redirect
from flask_restful import reqparse, abort, Api, Resource
from scraper.playstore.AppReviewScraper import AppReviewScraper
from graphs.playstore.AppReviewGraphRenderer import AppReviewGraphRenderer
import os
from dotenv import load_dotenv
from flask_cors import CORS, cross_origin
from scraper.appstore.AppStoreReviewScraper import AppStoreReviewScraper
load_dotenv()
import json

app = Flask(__name__)
cors = CORS(app, resources={r"*": {"origins": "*"}})
api = Api(app)
parser = reqparse.RequestParser()
parser.add_argument('key')
parser.add_argument('appname')
class SendGraphs(Resource):
    @cross_origin()
    def get(self):
        try:
            args = parser.parse_args()
            link = args['key']
            if 'play.google.com' in link:
                user_query = link[link.index('=')+1:]
                scraper = AppReviewScraper(app_id = user_query)
                scraper.get_reviews()
                renderer = AppReviewGraphRenderer(scraper.app_reviews_df)
                type = "playstore"
            elif 'apps.apple.com' in link:
                user_query = link[30:link.rindex('/')]
                scraper = AppStoreReviewScraper(app_name = user_query)
                scraper.get_reviews()
                renderer = AppReviewGraphRenderer(scraper.store_reviews_df)
                type="appstore"
            else:
                user_query = link
                scraper = ReviewScraper(key = user_query)
                reviews = scraper.get_reviews()
                renderer = GraphRenderer(reviews = reviews)

                big5_traits_graph, big5_traits_val_graph, emotional_traits_graph = renderer.drawBehavorialEmotionalChart()

                top_keywords = renderer.drawTopKeywords()

                star_dist = renderer.starDistribution()

                positive_reviews, negative_reviews = renderer.reviews_split()

                return {
                "type":"atlassian",
                "big5_traits_graph": big5_traits_graph,
                "big5_traits_val_graph" : big5_traits_val_graph,
                "emotional_traits_graph" : emotional_traits_graph,
                "star_dist" : star_dist,
                "top_keywords" : top_keywords,
                "positive_reviews" : positive_reviews,
                "negative_reviews" : negative_reviews
               }

            emotional_chart, behavorial_chart, iptc_chart = renderer.drawBehavorialEmotionalChart()
            top_keywords_chart = renderer.drawTopKeywords()
            rating_bar_chart = renderer.drawRatingHistogram()
            rating_line_chart = renderer.drawRatingLinePlot()
            positive_reviews, negative_reviews = renderer.reviews_split()
            return {
                "type":type,
                "emotional_chart" : emotional_chart,
                "behavorial_chart" : behavorial_chart,
                "iptc_chart" : iptc_chart,
                "top_keywords_chart" : top_keywords_chart,
                "rating_bar_chart" : rating_bar_chart,
                "rating_line_chart" : rating_line_chart,
                "positive_reviews" : positive_reviews,
                "negative_reviews" : negative_reviews
            }
        except Exception as e:
            return {"exception" : repr(e)}

api.add_resource(SendGraphs, '/analyze')

if __name__ == '__main__':
    app.run(debug=True)