Cheongwoon commited on
Commit
d377ca8
Β·
verified Β·
1 Parent(s): 7ed09e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -8,7 +8,7 @@ summarizer = pipeline("summarization", model="digit82/kobart-summarization", tok
8
 
9
  @app.route("/")
10
  def home():
11
- return "CW KoBART μš”μ•½ μ„œλ²„ μž‘λ™ 쀑 (λ©€ν‹° μ—”λ“œν¬μΈνŠΈ /summarize, /econalyze)"
12
 
13
  @app.route("/summarize", methods=["POST"])
14
  def summarize():
@@ -34,5 +34,16 @@ def econalyze():
34
  except Exception as e:
35
  return jsonify({"error": str(e)}), 500
36
 
 
 
 
 
 
 
 
 
 
 
 
37
  if __name__ == "__main__":
38
  app.run(host="0.0.0.0", port=7860)
 
8
 
9
  @app.route("/")
10
  def home():
11
+ return "CW KoBART μš”μ•½ μ„œλ²„ μž‘λ™ 쀑 (λ©€ν‹° μ—”λ“œν¬μΈνŠΈ /summarize, /econalyze, /ideas)"
12
 
13
  @app.route("/summarize", methods=["POST"])
14
  def summarize():
 
34
  except Exception as e:
35
  return jsonify({"error": str(e)}), 500
36
 
37
+ @app.route("/ideas", methods=["POST"])
38
+ def ideas():
39
+ data = request.get_json()
40
+ keywords = data.get("keywords", [])
41
+ if not keywords:
42
+ return jsonify({"ideas": []})
43
+ try:
44
+ return jsonify({"ideas": [f"{kw} κ΄€λ ¨ λΈ”λ‘œκ·Έ 아이디어" for kw in keywords]})
45
+ except Exception as e:
46
+ return jsonify({"error": str(e)}), 500
47
+
48
  if __name__ == "__main__":
49
  app.run(host="0.0.0.0", port=7860)