heysho commited on
Commit
8cb3f19
·
verified ·
1 Parent(s): cb4d53f

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +82 -0
app.py ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import traceback
2
+ import streamlit as st
3
+ from langchain_core.prompts import ChatPromptTemplate
4
+ from langchain_core.output_parsers import StrOutputParser
5
+ from langchain_openai import ChatOpenAI
6
+ from langchain_anthropic import ChatAnthropic
7
+ from langchain_google_genai import ChatGoogleGenerativeAI
8
+
9
+ ###### dotenv を利用する場合 ######
10
+ try:
11
+ from dotenv import load_dotenv
12
+ load_dotenv()
13
+ except ImportError:
14
+ import warnings
15
+ warnings.warn("dotenv not found. Please make sure to set your environment variables manually.", ImportWarning)
16
+ ################################################
17
+
18
+
19
+ PROMPT = """
20
+ ## 課題: 当社の製品「レビューアナライザー」の顧客レビューを分析します。購入を検討している可能性のある顧客に役立つ重要なポイントを抽出することに焦点を当ててください。全体的な顧客満足度を判断するために感情分析を含めます。
21
+ ## レビューサイト: Facebook、Instagram、その他の製品の市場存在に基づいた関連フォーラムからレビューを収集してください。
22
+ ## 目的: 現在のユーザーが製品についてどう思っているかについて、潜在的な顧客に明確で簡潔な要約を提供することです。最も称賛されている機能、一般的な問題、および一般的な感情を強調してください。
23
+ ## 追加: レビューの出典と分析した投稿数を示しながら、約200語で分析を要約します。要約は直截的で理解しやすく、購入決定を支援するために調整されていることを確認してください。
24
+ - ブランド: {brand},
25
+ - 製品:{product},
26
+ - 地域: {region}
27
+ """
28
+
29
+ def init_page():
30
+ st.set_page_config(
31
+ page_title="商品レビュー確認AIエージェント",
32
+ page_icon="🔍"
33
+ )
34
+ st.header("商品レビュー確認AIエージェント🔍")
35
+
36
+
37
+ def select_model(temperature=0):
38
+ models = ("GPT-4o","GPT-4o-mini", "Claude 3.5 Sonnet", "Gemini 1.5 Pro")
39
+ model_choice = st.radio("Choose a model:", models)
40
+ if model_choice == "GPT-4o":
41
+ return ChatOpenAI(temperature=temperature, model_name="gpt-4o")
42
+ elif model_choice == "GPT-4o-mini":
43
+ return ChatOpenAI(temperature=temperature, model_name="gpt-4o-mini")
44
+ elif model_choice == "Claude 3.5 Sonnet":
45
+ return ChatAnthropic(temperature=temperature, model_name="claude-3-5-sonnet-20240620")
46
+ elif model_choice == "Gemini 1.5 Pro":
47
+ return ChatGoogleGenerativeAI(temperature=temperature, model="gemini-1.5-pro-latest")
48
+
49
+ def init_chain():
50
+ llm = select_model()
51
+ prompt = ChatPromptTemplate.from_messages([
52
+ ("user", PROMPT),
53
+ ])
54
+ output_parser = StrOutputParser()
55
+ chain = prompt | llm | output_parser
56
+ return chain
57
+
58
+ def main():
59
+ init_page()
60
+ chain = init_chain()
61
+ if chain:
62
+ brand = st.text_input("ブランド名を入力してください(例:ナイキ)", key="brand")
63
+ product = st.text_input("製品名を入力してください(例:エアフォースワンシューズ)", key="product")
64
+ region = st.text_input("地域を入力してください(例:アメリカ)", key="region", value="USA")
65
+ if st.button("Submit"):
66
+ result = chain.stream({"brand": brand, "product": product, "region": region})
67
+ st.write(result)
68
+
69
+
70
+ if __name__ == '__main__':
71
+ main()
72
+
73
+ # Style adjustments (optional, remove if not needed)
74
+ st.markdown(
75
+ """
76
+ <style>
77
+ /* Custom style adjustments */
78
+ .st-emotion-cache-iiif1v { display: none !important; }
79
+ </style>
80
+ """,
81
+ unsafe_allow_html=True,
82
+ )