logic rewritten

app.py

@@ -1,33 +1,56 @@
-from flask import Flask, request
-from textblob import TextBlob
-
-flask_app = Flask(__name__)
-
-@flask_app.route("/text/summarize", methods=["POST"])
-def summarize():
-    text = request.json.get("text")
-    return {"summary": TextBlob(text).summarize()}
-
-@flask_app.route("/text/extract", methods=["POST"])
-def extract():
-    text = request.json.get("text")
-    return {"keywords": TextBlob(text).keywords}
-
-@flask_app.route("/text/paraphrase", methods=["POST"])
-def paraphrase():
-    text = request.json.get("text")
-    return {"paraphrase": TextBlob(text).correct()}
-
-@flask_app.route("/text/grammar", methods=["POST"])
-def grammar():
-    text = request.json.get("text")
-    return {"errors": TextBlob(text).spellcheck()}
-
-@flask_app.route("/chat", methods=["POST"])
-def chat():
-    prompt = request.json.get("prompt")
-    # TODO: create chatting logic
-    return {"response": f"prompt: {prompt}\nresponse: Still working on Text Generation logic, thanks for the patience!"}
-
-if __name__ == "__main__":
-    flask_app.run(host="0.0.0.0", port=7860)
+from flask import Flask, request, jsonify
+from nltk import sent_tokenize, word_tokenize, FreqDist
+from nltk.corpus import stopwords
+from nltk.tokenize.treebank import TreebankWordDetokenizer
+import nltk
+
+app = Flask(__name__)
+
+nltk.download("punkt")
+nltk.download("stopwords")
+
+stop_words = set(stopwords.words("english"))
+
+def summarize_text(text):
+    sentences = sent_tokenize(text)
+    words = word_tokenize(text)
+
+    # Remove stopwords
+    words = [word.lower() for word in words if word.isalnum() and word.lower() not in stop_words]
+
+    # Calculate word frequency
+    freq_dist = FreqDist(words)
+
+    # Sort sentences based on the sum of word frequencies
+    sorted_sentences = sorted(sentences, key=lambda sentence: sum(freq_dist[word] for word in word_tokenize(sentence)))
+
+    # Take the top 3 sentences as the summary
+    summary = " ".join(sorted_sentences[-3:])
+    
+    return summary
+
+def extract_keywords(text):
+    words = word_tokenize(text)
+    keywords = [word.lower() for word in words if word.isalnum() and word.lower() not in stop_words]
+    return keywords
+
+@app.route('/text/summarize', methods=['POST'])
+def summarize_text_route():
+    data = request.get_json()
+    text = data['text']
+    
+    summary = summarize_text(text)
+    
+    return jsonify({'summary': summary})
+
+@app.route('/text/extract', methods=['POST'])
+def extract_keywords_route():
+    data = request.get_json()
+    text = data['text']
+    
+    keywords = extract_keywords(text)
+    
+    return jsonify({'keywords': keywords})
+
+if __name__ == '__main__':
+    app.run(debug=True)