Create app.py

app.py

@@ -0,0 +1,637 @@
+import gradio as gr
+import re
+import hashlib
+import mysql.connector as mydb
+import time
+
+#import cohere
+import os
+from dotenv import load_dotenv
+
+import numpy as np
+import psycopg2
+from google import genai
+from google.genai import types
+
+import uuid
+import base64
+import requests
+import asyncio
+import random
+import pandas as pd
+from docx import Document
+from llama_parse import LlamaParse
+from llama_index.core import SimpleDirectoryReader
+
+load_dotenv(verbose=True)
+
+client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
+
+chat_history = []
+intreme = []
+
+basic_format = ""
+
+class DoSwitch:
+    def __init__(self):
+        # 状態（state）をインスタンス変数として保持
+        self.is_logged = False
+        self.first_exec = False
+        self.lastlog = ""
+
+    def log_on(self):
+        self.is_logged = True
+
+    def log_oout(self):
+        self.log_on = False
+
+    def first_executed(self):
+        self.first_exec = True
+
+my_switch = DoSwitch()
+
+def clean_text(text):
+    # 改行とタブを削除
+    return text.replace('\n', '').replace('\t', '').replace(' ', '')
+
+def update_file_type(file):
+    global filepath
+    filepath = file
+    if file is not None:
+        # アップロードされたファイルの拡張子を取得
+        _, ext = os.path.splitext(file.name)
+        return ext
+    return ""
+
+
+def respond(ctype, msg, username, response_check):
+    print("response_check:",response_check)
+    if response_check == True:
+        prompt = f"""「答え」をウェブで調べて、必ず、日本語で答えてください。
+        「答え」={msg}
+        """
+    else:
+        prompt = f"""以下の「レポート」を「基本形」に沿って正しく明記されているかをチェックし、レポートがどれ位「基本形」に沿った形でカバーしているかを教えてください。尚、漏れている項目を表示し、その内容をウェブから検索し、アドバイスしてください。最終的に内容がわかりやすかったか否かを教えてください。「基本形」にどれ位沿っているかも教えてください。必ず、日本語で答えてください。
+        「レポート」={msg}
+        「基本形」={basic_format}
+        """
+
+    gresponse = client.models.generate_content(
+        model="gemini-2.5-flash",
+        contents=[prompt]
+    )
+    summary = gresponse.text
+
+    print("summary1:",summary)
+    
+    return summary
+
+
+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 hash_password(password):
+    return hashlib.sha256(password.encode()).hexdigest()
+
+
+def login_user(username, password):
+    conn = mydb.connect(
+        host='www.ryhintl.com',
+        port='36000',
+        user='smairuser',
+        password='smairuser',
+        database='smair'
+    )
+    c = conn.cursor()
+    hashed_pw = hash_password(password)
+    sqlcmd = "SELECT * FROM gyoseki_users WHERE username = '"+username+"' AND password = '"+hashed_pw+"'"
+    c.execute(sqlcmd)
+    
+    user = c.fetchone()
+    conn.close()
+
+    if user:
+        return True, user
+    return False, "ユーザー名またはパスワードが無効です"
+
+
+def process_logout():
+    time.sleep(1)
+    return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+
+# 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],
+                "basic_format": result[5]
+            }
+
+            uid = result[5]
+        
+            conn = mydb.connect(
+                host='www.ryhintl.com',
+                port='36000',
+                user='smairuser',
+                password='smairuser',
+                database='smair'
+            )
+            c = conn.cursor()
+    
+            sqlcmd = "SELECT contents FROM gyoseki_basic_format WHERE id = "+str(uid)
+            c.execute(sqlcmd)
+    
+            content = c.fetchone()
+            global basic_format
+            basic_format = content
+            conn.close()
+
+            return gr.update(value="<p class='success-message'></p>"), True, username, user_info, gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), gr.update(visible=False)
+            #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(visible=False), gr.update(visible=True)
+            return gr.update(value=f"<p class='error-message'>{result}</p>"), False, None, None, gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
+            #return gr.update(value=f"<p class='error-message'>{result}</p>"), False, None, None, gr.update(visible=True), gr.update(visible=False)
+
+
+def filecomp_change(prompt):
+    if prompt == None:
+        return gr.update(visible=False),gr.update(visible=True),gr.update(visible=True)
+    else:
+        return gr.update(visible=True),gr.update(visible=True),gr.update(visible=False)
+    
+def summary_change(prompt):
+    if len(prompt) == 0:
+        return gr.update(visible=False)
+    else:
+        return gr.update(visible=True)
+
+
+def prompt_change(prompt):
+    if len(prompt["files"]) > 0:
+        print("cnt:",len(prompt["files"]))
+        for x in prompt["files"]:
+            file_path = prompt['files'][0]
+            print("file path",file_path)
+        return gr.update(visible=False),gr.update(visible=True),gr.update(visible=True)
+    else:
+        print("prompt:",prompt["text"])
+        return gr.update(visible=True),gr.update(visible=False),gr.update(visible=True)
+
+
+def show_tab_1():
+    return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
+
+def show_tab_2():
+    return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
+
+def show_tab_3():
+    return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
+
+def embed(texts, input_type):
+    # 改行または句点で分割（任意で調整可能）
+    if isinstance(texts, str):
+        inputs = [t.strip() for t in texts.split("\n") if t.strip()]
+    else:
+        inputs = texts
+
+    result = client.models.embed_content(
+        model="gemini-embedding-001",
+        contents=inputs,
+        config=types.EmbedContentConfig(output_dimensionality=1024)
+    )
+
+    # 結果を整形して表示
+    formatted = ""
+    for i, embedding in enumerate(result.embeddings):
+        #formatted += f"Embedding (Array{i}: {embedding.values} \n\n"
+        formatted += f"{embedding.values}"
+    return formatted
+
+
+def summarize_text(long_text, username, summarize_text):
+    
+    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
+
+    if (summarize_text == '保存'):
+
+        response = embed(summary,"search_document")
+
+        embedding_str = response
+
+        conn = psycopg2.connect(
+            dbname="smair",
+            user="smairuser",
+            password="smairuser",
+            host="www.ryhintl.com",
+            port=10629
+        )
+
+        cur = conn.cursor()
+        sql = "INSERT INTO dailog_logs (userid, content, embedding) VALUES (%s, %s, %s)"
+        cur.execute(sql, (username, summary, f"{embedding_str}"))
+        conn.commit()
+        cur.close()
+        conn.close()
+        return summary
+    else:
+        return summary
+
+
+def generate_summary(input_text, username, summary_save):
+        if not username:
+            return "ログインしてください。"
+        
+        summary = summarize_text(input_text, username, summary_save)
+        return summary
+
+def tnews():
+    res = client.models.generate_content(
+        model="gemini-2.5-flash",
+        contents=["東証のtopixに関するニュースを教えてください。必ず、日本語で答えてください。"]
+    )
+    return res.text
+
+
+def generate_response(dtype, file_name, file_type, msg, username, report_check):
+             
+        #if (file_name == None):  # Should not happen if UI is correctly managed
+        if (msg == '' and file_name == ''):  # Should not happen if UI is correctly managed
+            ret = []
+            ret.append(["レポートをアップロードしてください。","Please Upload report."])
+            yield ret
+        elif (msg == '' and file_name != ''):
+            
+            if (my_switch.first_exec == True):
+                intreme.append(["レポートを処理中...",tnews()])
+                yield intreme
+
+                parser = LlamaParse(
+                    api_key=os.environ.get("LLAMA_API_KEY"),
+                    language="ja",
+                    high_res_ocr=True,
+                    user_prompt="提供された文書はスライドである。1ページにはファイルのタイトル、2ページは目次となっており、それ以降のページは各ページごとの副題を持つ。資料は、テキスト、図、表、グラフ、イラストを含む。",
+                    result_type="markdown"  # "markdown" または "text" が選択可能
+                )
+    
+                file_extractor = {file_type: parser}
+                documents = SimpleDirectoryReader(input_files=[file_name], file_extractor=file_extractor, recursive=False).load_data()
+    
+                print("documents:",documents)
+                texts = [doc.text for doc in documents]
+                print("texts:",texts)
+        
+                combined_text = clean_text(texts[0])
+                msg = texts[0]
+
+                bot_message = respond(file_type, msg, username, report_check)
+                print("botmsg:",bot_message)
+                chat_history.append([msg, bot_message])
+                intreme.clear()
+                yield chat_history
+            else:
+                #gotresp = requests.get('')
+                conn = psycopg2.connect(
+                    dbname="smair",
+                    user="smairuser",
+                    password="smairuser",
+                    host="www.ryhintl.com",
+                    port=10629
+                )
+
+                cur = conn.cursor()
+                sql = "select content from dailog_logs where userid = '"+username+"'"
+                cur.execute(sql)
+                plog = cur.fetchone()
+                conn.close()
+                
+                my_switch.first_exec = True
+                
+                my_switch.lastlog = plog
+                prompt = f"""「前回の会話ログ」でユーザーの予測や仮説などから結果を核にする項目があれば、確認する質問を列挙してください。例えば、ハイブリッド車の販売が堅調でコスト削減効果が期待できることに言及したが、具体的な過去のコストや利益率の推移データは提示せず、トヨタ自動車の公式財務報告書などを参照するよう促した。などについてハイブリッド車の販売好調がコスト増加抑制に寄与しているという当初の推測について、その後、何か追加で確認できた点はありますか？」のように質問を列挙してください。
+                「前回の会話ログ」={my_switch.lastlog}
+                """
+                gresponse = client.models.generate_content(
+                    model="gemini-2.5-flash",
+                    contents=[prompt]
+                )
+                lastqa = gresponse.text
+                msg = lastqa
+
+                intreme.append(["レポートを処理中...",lastqa])
+                #yield intreme
+
+                '''parser = LlamaParse(
+                    api_key=os.environ.get("LLAMA_API_KEY"),
+                    language="ja",
+                    high_res_ocr=True,
+                    user_prompt="提供された文書はスライドである。1ページにはファイルのタイトル、2ページは目次となっており、それ以降のページは各ページごとの副題を持つ。資料は、テキスト、図、表、グラフ、イラストを含む。",
+                    result_type="markdown"  # "markdown" または "text" が選択可能
+                )
+    
+                file_extractor = {file_type: parser}
+                documents = SimpleDirectoryReader(input_files=[file_name], file_extractor=file_extractor, recursive=False).load_data()
+    
+                texts = [doc.text for doc in documents]
+        
+                combined_text = clean_text(texts[0])
+                msg = texts[0]'''
+
+                #bot_message = respond(file_type, msg, username, report_check) + "\n" + lastqa
+                bot_message = respond(file_type, msg, username, report_check)
+                chat_history.append([msg, bot_message])
+                intreme.clear()
+                yield chat_history
+
+        else:
+            intreme.append(["レポートを処理中...",tnews])
+            yield intreme
+
+            bot_message = respond(file_type, msg, username, report_check)
+            chat_history.append([msg, bot_message])
+            intreme.clear()
+            yield chat_history
+
+
+
+
+def summary_change(prompt):
+    if len(prompt) == 0:
+        return gr.update(visible=False)
+    else:
+        return gr.update(visible=True)
+    
+
+
+
+
+with gr.Blocks(title="Fund Manager Academic 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;
+        /*width: 100%;
+        height: 100%;
+        background-image: url('https://images.unsplash.com/photo-1527181152855-fc03fc7949c8?auto=format&w=1000&dpr=2');
+        background-size: cover;
+        background-repeat: no-repeat;
+        background-position: center center;*/
+    }
+    .main {
+        background-color: #f8f9fa;
+    }
+    .logo-container {
+        position: absolute;
+        top: 1px;
+        left: 20px;
+        z-index: 1000;
+    }
+    .logo-container img {
+        height: 30px;
+        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;
+    }
+               
+    .rcard {
+        background: #000055;
+        border-radius: 15px;
+        padding: 2rem;
+        box-shadow: 0 10px 20px rgba(0,0,0,0.1);
+        margin-bottom: 1rem;
+        height: 650px;
+        overflow: hidden;
+    }
+               
+    .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: 1rem;
+        color: #95a5a6;
+        font-size: 0.8rem;
+    }
+    .avatar {
+        width: 50px;
+        height: 50px;
+        border-radius: 50%;
+        margin: 0 auto 1rem auto;
+        display: block;
+        object-fit: cover;
+        border: 3px solid #4a90e2;
+    }
+    
+    .avatar:hover {
+        transform: scale(1.2);
+        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;
+    }
+    .my-custom-button {
+        background-color: transparent !important; /* 緑色 */
+        color: black !important;
+        padding: 10px 20px !important;
+        border: solid thin black !important;
+        border-radius: 5px !important;
+        cursor: pointer !important;
+        font-size: 16px !important;
+        width: 100%;
+        transition: background-color 0.3s ease !important;
+    }
+
+    .my-custom-button:hover {
+        background-color: #fff5ee !important; /* ホバー時の色 */
+    }
+""") as demo:
+    # State variables
+    current_username = gr.State(None)
+    current_user_info = gr.State(None)
+    logged_in_state = gr.State(False)
+
+    with gr.Sidebar(position="left", open=True):
+        gr.Markdown("©️ FUND MANAGER BUDDY 2025")
+        signin = gr.Button("サインイン", visible=True, elem_classes=["my-custom-button"], icon="https://img.icons8.com/fluency/48/azure-1.png")
+        query = gr.Button("クエリー", visible=False, elem_classes=["my-custom-button"], icon="https://img.icons8.com/fluency/48/orthogonal-view.png")
+        smry = gr.Button("ダイアログ・サマリー", visible=False, elem_classes=["my-custom-button"], icon="https://img.icons8.com/3d-fluency/48/domain.png")
+
+    #with gr.Tabs() as tabs:
+        # Set the first tab to be visible by default
+    with gr.Column("タブ1", visible=False, elem_classes="card") as b1:
+        signup_btn = gr.Button("サインアップ", elem_classes=["my-custom-button"], icon="https://img.icons8.com/color/48/000000/google-logo.png")
+        login_username = gr.Textbox(label="ユーザー名", placeholder="ユーザー名を入力してください")
+        login_password = gr.Textbox(label="パスワード", type="password", placeholder="パスワードを入力してください")
+        login_status = gr.Markdown("")
+        login_btn = gr.Button("サインイン")
+
+        login_btn.click(
+            process_login,
+            inputs=[login_username, login_password, current_user_info],
+            outputs=[login_status, logged_in_state, current_username, current_user_info, signin, query, smry, b1]
+        )
+        
+    with gr.Column("タブ2", visible=False) as b2:
+        with gr.TabItem("クエリー"):
+            logout_btn = gr.Button("ログアウト")
+            dialog_ctype = gr.Dropdown(["デフォルト"], label="会話形式", value="デフォルト")
+
+            with gr.Row(visible=True) as filecomp:
+                file_input = gr.File(label="レポートをアップロード(PDF,pptx,xlsx,docx,txt,mdなど)")
+                file_type = gr.Textbox(label="ファイルタイプ")
+
+            with gr.Row(visible=True) as reportcomp:
+                report_check = gr.Checkbox(label="レポート質問", info="レポート質問に対する答えの時にチェックしてください。")
+                report_content = gr.Textbox(label="レポート", lines=10, show_copy_button=True)
+                '''report_content = gr.MultimodalTextbox(
+                    interactive=True,
+                    file_count="single",
+                    placeholder="レポートの内容や質問を入力してください",
+                    label="レポート",
+                    lines=10,
+                    show_label=True,
+                    sources=["upload", "microphone"],
+                )'''
+
+        dialog_output = gr.Chatbot(label="対話履歴", height=400, show_copy_all_button=True)
+        dialog_submit_btn = gr.Button("生成", visible=True)
+
+        file_input.change(fn=update_file_type, inputs=file_input, outputs=file_type)
+
+        #file_input.change(
+            #filecomp_change,
+            #inputs=[file_input],
+            #outputs=[dialog_submit_btn, filecomp, reportcomp]
+        #)
+
+        report_content.change(
+            prompt_change,
+            #handle_change,
+            inputs=[report_content],
+            outputs=[dialog_submit_btn, filecomp, reportcomp]
+        )
+
+        dialog_submit_btn.click(
+            generate_response,
+            inputs=[dialog_ctype, file_input, file_type, report_content, current_username, report_check],
+            outputs=[dialog_output]
+        )
+
+    with gr.Column("タブ3", visible=False) as b3:
+        with gr.TabItem("ダイアログ・サマリー"):
+            summary_input = gr.Textbox(label="要約したいユーザー・ダイアログを入力", lines=5)
+            summary_save = gr.Dropdown(["保存","保存しない"], label="DB保存", value="保存しない")
+            summary_output = gr.Markdown(label="要約結果", height=400, show_copy_button=True)
+            summary_submit_btn = gr.Button("要約", visible=False)
+
+            summary_input.change(
+                summary_change,
+                inputs=[summary_input],
+                outputs=[summary_submit_btn]
+            )
+
+            summary_submit_btn.click(
+                generate_summary,
+                inputs=[summary_input, current_username, summary_save],
+                outputs=summary_output
+            )
+
+    signin.click(show_tab_1, inputs=None, outputs=[b1, b2, b3])
+    query.click(show_tab_2, inputs=None, outputs=[b1, b2, b3])
+    smry.click(show_tab_3, inputs=None, outputs=[b1, b2, b3])
+    logout_btn.click(
+        process_logout,
+        inputs=[],
+        outputs=[signin,query,smry,b2]
+    )
+    #with gr.Sidebar(position="left"):
+        #gr.Markdown("## サイドバー")
+        #gr.Button("サインイン", elem_classes=["my-custom-button"], icon="https://img.icons8.com/fluency/48/azure-1.png").click(show_tab_1, inputs=None, outputs=[b1, b2, b3])
+        #query = gr.Button("クエリー", visible=False, elem_classes=["my-custom-button"], icon="https://img.icons8.com/fluency/48/orthogonal-view.png").click(show_tab_2, inputs=None, outputs=[b1, b2, b3])
+        #smry = gr.Button("ダイアログ・サマリー", visible=False, elem_classes=["my-custom-button"], icon="https://img.icons8.com/3d-fluency/48/domain.png").click(show_tab_3, inputs=None, outputs=[b1, b2, b3])
+
+demo.launch()