Create app.py

app.py

@@ -0,0 +1,438 @@
+import gradio as gr
+import re
+import hashlib
+import sqlite3
+import time
+
+import cohere
+import os
+from dotenv import load_dotenv
+
+import numpy as np
+import psycopg2
+from google import genai
+
+import uuid
+import base64
+import requests
+import asyncio
+import random
+import pandas as pd
+
+load_dotenv(verbose=True)
+
+# Initialize Qdrant and Cohere clients
+co = cohere.ClientV2(api_key=os.environ.get("COHERE_API_KEY"))
+client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
+
+# Database Initialization and Utility Functions
+def init_db():
+    conn = sqlite3.connect('auth.db')
+    c = conn.cursor()
+    c.execute('''
+        CREATE TABLE IF NOT EXISTS users (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            username TEXT UNIQUE NOT NULL,
+            email TEXT UNIQUE NOT NULL,
+            password TEXT NOT NULL,
+            phone TEXT,
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+        )
+    ''')
+    conn.commit()
+    conn.close()
+
+init_db()
+
+def hash_password(password):
+    return hashlib.sha256(password.encode()).hexdigest()
+
+def validate_username(username):
+    if len(username) < 4:
+        return "ユーザー名は4文字以上である必要があります"
+    if not re.match("^[a-zA-Z0-9_]+$", username):
+        return "ユーザー名には文字、数字、アンダースコアのみ使用できます"
+    return None
+
+def validate_email(email):
+    if not re.match(r"^[a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$", email):
+        return "有効なメールアドレスを入力してください"
+    return None
+
+def validate_password(password):
+    if len(password) < 8:
+        return "パスワードは8文字以上でなければなりません"
+    if not any(char.isdigit() for char in password):
+        return "パスワードには少なくとも1つの数字を含める必要があります"
+    if not any(char.isupper() for char in password):
+        return "パスワードには少なくとも1つの大文字を含める必要があります"
+    return None
+
+def validate_phone(phone):
+    if phone and not re.match(r"^\+?[0-9\s\-]+$", phone):
+        return "有効な電話番号を入力してください"
+    return None
+
+def register_user(username, email, password, phone):
+    conn = sqlite3.connect('auth.db')
+    c = conn.cursor()
+    c.execute("SELECT * FROM users WHERE username = ? OR email = ?", (username, email))
+    if c.fetchone():
+        conn.close()
+        return False, "ユーザー名またはメールアドレスは既に存在します"
+
+    hashed_pw = hash_password(password)
+    c.execute(
+        "INSERT INTO users (username, email, password, phone) VALUES (?, ?, ?, ?)",
+        (username, email, hashed_pw, phone))
+    conn.commit()
+    conn.close()
+    return True, "登録が成功しました"
+
+def login_user(username, password):
+    conn = sqlite3.connect('auth.db')
+    c = conn.cursor()
+    hashed_pw = hash_password(password)
+
+    c.execute("SELECT * FROM users WHERE username = ? AND password = ?", (username, hashed_pw))
+    user = c.fetchone()
+    conn.close()
+
+    if user:
+        return True, user
+    return False, "ユーザー名またはパスワードが無効です"
+
+# Cohere and Gemini Helper Functions
+def embed(input, input_type):
+    response = co.embed(texts=input, model='embed-multilingual-v3.0', input_type=input_type, embedding_types=['ubinary'])
+    return [np.unpackbits(np.array(embedding, dtype=np.uint8)) for embedding in response.embeddings.ubinary]
+
+def summarize_text(long_text, username):
+    if not long_text:
+        return "要約するテキストがありません。"
+
+    prompt = f"次の文章をuserとassistantに分けて的確に要約してください:\n{long_text}"
+    gresponse = client.models.generate_content(
+        model="gemini-2.5-flash",
+        contents=[prompt]
+    )
+    summary = gresponse.text
+    return summary
+
+def respond(ctype, msg, username):
+    conn = psycopg2.connect(
+        dbname="smair",
+        user="smairuser",
+        password="smairuser",
+        host="www.ryhintl.com",
+        port=10629
+    )
+    cur = conn.cursor()
+
+    query_embedding = embed([msg], 'search_query')[0].tolist()
+
+    cur.execute(
+        'SELECT content, 1 - (embedding <=> %s::vector) AS similarity FROM dailog_logs WHERE (1 - (embedding <=> %s::vector)) <> 0 ORDER BY similarity ASC',
+        (query_embedding, query_embedding)
+    )
+    proof = [row[0] for row in cur.fetchall()]
+
+    cur.execute(
+        'SELECT content, 1 - (embedding <=> %s::vector) AS similarity FROM monitoring_dialog WHERE (1 - (embedding <=> %s::vector)) <> 0 ORDER BY similarity ASC',
+        (query_embedding, query_embedding)
+    )
+    vector_resp = [row[0] for row in cur.fetchall()]
+
+    if ctype == "デフォルト":
+        message = f"{vector_resp}に基づいて{proof}を交えながら{msg}に対する答えを正確に出力してください。"
+    else:
+        message = f"{vector_resp}に基づいて{proof}を交えながら{msg}に対する答えを正確に関西弁で出力してください。"
+
+    messages = [
+        {"role": "system", "content": "あなたは、優秀なアシスタントです。"},
+        {"role": "user", "content": message},
+    ]
+
+    response = co.chat(model="command-a-03-2025", messages=messages)
+    bot_message = response.message.content[0].text
+    
+    cur.close()
+    conn.close()
+    return bot_message
+
+# Gradio UI with Blocks
+with gr.Blocks(title="Fund Manager Buddy", css="""footer {visibility: hidden;} #header {display: flex; justify-content: space-between; align-items: center; font-size: 24px; font-weight: bold;} #logo {width: 50px; height: 50px;} 
+    .gradio-container {
+        background-color: #f8f9fa;
+    }
+    .main {
+        background-color: #f8f9fa;
+    }
+    .logo-container {
+        position: absolute;
+        top: 20px;
+        left: 20px;
+        z-index: 1000;
+    }
+    .logo-container img {
+        height: 50px;
+        width: auto;
+    }
+    .title {
+        font-size: 1.5rem;
+        font-weight: 700;
+        color: #2c3e50;
+        text-align: center;
+        margin-bottom: 1.5rem;
+    }
+    .subtitle {
+        font-size: 0.1rem;
+        color: #7f8c8d;
+        text-align: center;
+        margin-bottom: 2rem;
+    }
+    .card {
+        background: white;
+        border-radius: 15px;
+        padding: 2rem;
+        box-shadow: 0 10px 20px rgba(0,0,0,0.1);
+        margin-bottom: 1rem;
+    }
+    .success-message {
+        color: #27ae60;
+        text-align: center;
+        margin-top: 1rem;
+    }
+    .error-message {
+        color: #e74c3c;
+        text-align: center;
+        margin-top: 1rem;
+    }
+    .footer {
+        text-align: center;
+        margin-top: 2rem;
+        color: #95a5a6;
+        font-size: 0.5rem;
+    }
+    .avatar {
+        width: 100px;
+        height: 100px;
+        border-radius: 50%;
+        margin: 0 auto 1rem auto;
+        display: block;
+        object-fit: cover;
+        border: 3px solid #4a90e2;
+    }
+    .gr-button {
+        width: 100%;
+        border-radius: 10px;
+        padding: 10px;
+        background-color: #4a90e2;
+        color: white;
+        border: none;
+        font-weight: 500;
+        transition: all 0.3s;
+    }
+    .gr-button:hover {
+        background-color: #357abd;
+        color: lightyellow;
+        transform: translateY(-2px);
+        box-shadow: 0 4px 8px rgba(0,0,0,0.1);
+    }
+    .gr-textinput, .gr-textbox {
+        border-radius: 10px;
+        padding: 10px;
+        border: 1px solid #ced4da;
+    }
+""") as buddy:
+    
+    gr.HTML('<div id="header"><span>🛡️ FUND MANAGER BUDDY</span><img id="logo" src="https://www.ryhintl.com/images/ryhlogo/ryhlogo.png" width="64" height="64" alt="Logo"></div>')
+    gr.Markdown("꧁ FM Buddy ꧂  ベクターDBに保存されている知識ベースのインベントリを使用してFMへの情報を共有します。")
+
+    # State variables
+    current_username = gr.State(None)
+    current_user_info = gr.State(None)
+    logged_in_state = gr.State(False)
+
+    with gr.Column(elem_classes="main"):
+        gr.HTML('<div class="logo-container"><img src="https://www.ryhintl.com/images/ryhlogo/ryhlogo.png" alt="Logo"></div>')
+
+        # Login/Register Section
+        with gr.Column(visible=True, elem_id="login_register_section") as login_register_section:
+            with gr.Tab("ログイン"):
+                gr.HTML('<h5 class="title">お帰りなさい！</h5>')
+                gr.HTML('<p style="text-align: center; margin-top: 1rem; font-size: 13px;">アカウントにアクセスするにはサインインしてください</p>')
+
+                with gr.Column(elem_classes="card"):
+                    login_username = gr.Textbox(label="ユーザー名", placeholder="ユーザー名を入力してください")
+                    login_password = gr.Textbox(label="パスワード", type="password", placeholder="パスワードを入力してください")
+                    login_status = gr.Markdown("")
+                    login_btn = gr.Button("サインイン")
+
+            with gr.Tab("アカウントを作成"):
+                gr.HTML('<h1 class="title">アカウントを作成</h1>')
+                gr.HTML('<p class="subtitle">今すぐ、参加して始めましょう！</p>')
+
+                with gr.Column(elem_classes="card"):
+                    reg_username = gr.Textbox(label="ユーザー名", placeholder="ユーザー名を選択してください")
+                    reg_email = gr.Textbox(label="電子メール", placeholder="メールアドレスを入力してください")
+                    reg_phone = gr.Textbox(label="電話番号 (オプション)", placeholder="+81-1234567890")
+                    reg_password = gr.Textbox(label="パスワード", type="password", placeholder="パスワードを入力してください")
+                    reg_confirm_password = gr.Textbox(label="パスワード再確認", type="password", placeholder="パスワードを再度入力してください")
+                    register_status = gr.Markdown("")
+                    register_btn = gr.Button("登録")
+
+        # Dashboard Section
+        with gr.Column(visible=False, elem_id="dashboard_section") as dashboard_section:
+            welcome_message = gr.Markdown()
+            gr.HTML('<p style="text-align: center; margin-top: 1rem; font-size: 10px;"">あなたは現在、アカウントにログインしています</p>')
+
+            user_avatar = gr.HTML()
+
+            with gr.Row():
+                with gr.Column(scale=0.1):
+                    gr.Markdown("### QUERY EXAMPLES")
+                    gr.Markdown("""
+                    - **トヨタ自動車の株を買いたいと思ってるんだけど、どう思いますか？**
+                    - **トヨタの株、買おかな思てんねんけど、どう思う？それと、買うときに参考にしたらええ判断材料とか基準、教えてくれへん？**
+                    - **トヨタ自動車の株価の上昇余地はどれくらいだと思いますか？現在の株価は¥2,508で、証券街のEPS予測（¥241.37）に対してPERは20倍です。**
+                    - **トヨタ自動車の過去の予測履歴があれば教えてください。**
+                    - **トヨタ自動車のハイブリッド自動車の販売が予想以上に好調でコスト増加が抑えられていると思われるが、過去のコストや利益率の推移を確認できますか？**
+                    """)
+                with gr.Column(scale=3):
+                    with gr.Tabs():
+                        with gr.TabItem("ダイアログ"):
+                            dialog_ctype = gr.Dropdown(["デフォルト", "関西弁"], label="会話形式", value="デフォルト")
+                            dialog_query = gr.Textbox(label="クエリーを入力してください。", lines=5, value="トヨタ自動車の株を買いたいと思ってるんだけど、どう思いますか？ 尚、購入のための参考にすべき判断材料や基準を教えてください")
+                            dialog_output = gr.Chatbot(label="対話履歴", bubble_full_width=False, height=400)
+                            dialog_submit_btn = gr.Button("生成")
+
+                        with gr.TabItem("ダイアログ・サマリー"):
+                            summary_input = gr.Textbox(label="要約したいユーザー・ダイアログを入力", lines=5)
+                            summary_output = gr.Markdown(label="要約結果")
+                            summary_submit_btn = gr.Button("要約")
+
+            logout_btn = gr.Button("ログアウト")
+
+    gr.HTML('<p class="footer">© 2025 Fund Manager Buddy. All rights reserved.</p>')
+
+    # Gradio Event Handlers
+
+    # Login Logic
+    def process_login(username, password, current_user_info_state):
+        username_error = validate_username(username)
+        password_error = validate_password(password)
+
+        if username_error:
+            return gr.update(value=f"<p class='error-message'>{username_error}</p>"), False, None, None, gr.update(visible=True), gr.update(visible=False)
+        elif password_error:
+            return gr.update(value=f"<p class='error-message'>{password_error}</p>"), False, None, None, gr.update(visible=True), gr.update(visible=False)
+        else:
+            success, result = login_user(username, password)
+            if success:
+                user_info = {
+                    "id": result[0],
+                    "username": result[1],
+                    "email": result[2],
+                    "phone": result[4]
+                }
+                return gr.update(value="<p class='success-message'></p>"), True, username, user_info, gr.update(visible=False), gr.update(visible=True)
+            else:
+                return gr.update(value=f"<p class='error-message'>{result}</p>"), False, None, None, gr.update(visible=True), gr.update(visible=False)
+
+    login_btn.click(
+        process_login,
+        inputs=[login_username, login_password, current_user_info],
+        outputs=[login_status, logged_in_state, current_username, current_user_info, login_register_section, dashboard_section]
+    )
+
+    # Register Logic
+    def process_register(username, email, phone, password, confirm_password):
+        errors = []
+        username_error = validate_username(username)
+        email_error = validate_email(email)
+        password_error = validate_password(password)
+        phone_error = validate_phone(phone)
+
+        if username_error:
+            errors.append(username_error)
+        if email_error:
+            errors.append(email_error)
+        if password_error:
+            errors.append(password_error)
+        if phone_error:
+            errors.append(phone_error)
+        if password != confirm_password:
+            errors.append("パスワードが一致しません。")
+
+        if errors:
+            error_html = "".join([f"<p class='error-message'>{e}</p>" for e in errors])
+            return gr.update(value=error_html), gr.update(visible=True), gr.update(visible=False)
+        else:
+            success, message = register_user(username, email, password, phone)
+            if success:
+                return gr.update(value=f"<p class='success-message'>{message}</p>"), gr.update(visible=True), gr.update(visible=False)
+            else:
+                return gr.update(value=f"<p class='error-message'>{message}</p>"), gr.update(visible=True), gr.update(visible=False)
+
+    register_btn.click(
+        process_register,
+        inputs=[reg_username, reg_email, reg_phone, reg_password, reg_confirm_password],
+        outputs=[register_status, login_register_section, dashboard_section]
+    )
+
+
+    # Update Dashboard on Login Success
+    def update_dashboard_ui(is_logged_in, username, user_info):
+        if is_logged_in:
+            welcome_msg = f'<h3 class="title">ようこそ！ {username} 様!</h3>'
+            avatar_html = f'<img src="https://ui-avatars.com/api/?name=' + username + '&background=4a90e2&color=fff&size=200" class="avatar">'
+            return welcome_msg, avatar_html, gr.update(visible=False), gr.update(visible=True)
+        return "", "", gr.update(visible=True), gr.update(visible=False)
+
+    logged_in_state.change(
+        update_dashboard_ui,
+        inputs=[logged_in_state, current_username, current_user_info],
+        outputs=[welcome_message, user_avatar, login_register_section, dashboard_section]
+    )
+
+    # Logout Logic
+    def process_logout():
+        time.sleep(1)
+        return False, None, None, gr.update(visible=True), gr.update(visible=False)
+
+    logout_btn.click(
+        process_logout,
+        inputs=[],
+        outputs=[logged_in_state, current_username, current_user_info, login_register_section, dashboard_section]
+    )
+
+    # Chat Response Logic
+    def generate_response(ctype, msg, chat_history, username):
+        if not username:  # Should not happen if UI is correctly managed
+            return chat_history, gr.update(value="ログインしてください。")
+
+        bot_message = respond(ctype, msg, username)
+        chat_history.append([msg, bot_message])
+        return chat_history, "" # Clear the input after sending
+
+    dialog_submit_btn.click(
+        generate_response,
+        inputs=[dialog_ctype, dialog_query, dialog_output, current_username],
+        outputs=[dialog_output, dialog_query]
+    )
+
+    # Summarize Logic
+    def generate_summary(input_text, username):
+        if not username:
+            return "ログインしてください。"
+        summary = summarize_text(input_text, username)
+        return summary
+
+    summary_submit_btn.click(
+        generate_summary,
+        inputs=[summary_input, current_username],
+        outputs=summary_output
+    )
+
+buddy.launch(favicon_path="favicon.ico", show_api=False)