カレンダーアップと統合

app/app.py

@@ -6,6 +6,9 @@ from dotenv import load_dotenv
 import requests
 from typing import Optional 
 from fastapi import FastAPI, Request, Header, HTTPException
+import sys
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
+from calendar_tool import calendar_main
 # .env 파일에서 환경 변수를 불러옵니다.
 load_dotenv()
 app = FastAPI()
@@ -64,11 +67,17 @@ async def handle_chatwork_webhook(
         message_id = event.get("message_id")
         
         message_body = event.get("body")
+        calendar_data = ""
         if message_body:
             # strip()을 사용하여 앞뒤 공백 및 줄바꿈 문자를 제거하고 출력합니다.
             cleaned_message = message_body.strip()
-            print(f"Received message: {cleaned_message}")
-            print(f"Received body: {message_body}")
+            if cleaned_message[0] != '会':
+                print("会議室の検索コマンドではありません。")
+                return
+            if cleaned_message:
+                calendar_data = calendar_main(cleaned_message)
+            if not calendar_data:
+                return
 
         if room_id and from_account_id and message_id:
             # Chatwork API 엔드포인트
@@ -81,7 +90,7 @@ async def handle_chatwork_webhook(
             # [rp] 태그를 사용하면 특정 메시지에 대한 답장 형식으로 보낼 수 있습니다.
             reply_body = (
                 f"[rp aid={from_account_id} to={room_id}-{message_id}]"
-                "\n정상적으로 수신되었습니다. ✅"
+                f"\n✅{calendar_data}"
             )
             
             payload = {"body": reply_body}

app/calendar_tool.py

@@ -0,0 +1,240 @@
+import os
+import sys
+import datetime
+import json
+import pytz
+from google.oauth2 import service_account
+from googleapiclient.discovery import build
+from google.auth.transport.requests import Request
+from google.oauth2.credentials import Credentials
+from google_auth_oauthlib.flow import InstalledAppFlow
+import google.generativeai as genai
+from vertexai.generative_models import GenerativeModel
+from dotenv import load_dotenv
+
+load_dotenv()
+base_path = os.path.dirname(__file__)
+credentials_path = os.path.join(base_path, 'new_credential.json')
+token_path = os.path.join(base_path, 'new_token.json')
+
+if not os.path.exists(credentials_path):
+    print(f"'{credentials_path}' 파일이 없어 새로 생성합니다.")
+    credentials_str = os.getenv('GOOGLE_CREDENTIALS_JSON')
+    if credentials_str:
+        with open(credentials_path, 'w', encoding='utf-8') as f:
+            f.write(credentials_str)
+    else:
+        print("경고: .env 파일에 'GOOGLE_CREDENTIALS_JSON' 변수가 없습니다.")
+
+if not os.path.exists(token_path):
+    print(f"'{token_path}' 파일이 없어 새로 생성합니다.")
+    token_str = os.getenv('GOOGLE_TOKEN_JSON')
+    if token_str:
+        with open(token_path, 'w', encoding='utf-8') as f:
+            f.write(token_str)
+
+
+CREDENTIALS_FILENAME = credentials_path
+TOKEN_FILENAME = token_path
+
+# Google Calendar API 스코프 (읽기 전용)
+SCOPES = ['https://www.googleapis.com/auth/calendar.readonly']
+# 기준 시간대 (GAS의 Session.getScriptTimeZone()에 해당)
+# 필요에 따라 'Asia/Tokyo', 'America/New_York' 등으로 변경하세요.
+TIMEZONE = 'Asia/Tokyo'
+
+rooms = [
+    {"capacity": 6, "name": "1F2", "email": os.getenv("GOOGLE_CALENDAR_1F2")},
+    {"capacity": 6, "name": "1F3", "email": os.getenv("GOOGLE_CALENDAR_1F3")},
+    {"capacity": 12, "name": "3F1", "email": os.getenv("GOOGLE_CALENDAR_3F1")},
+    {"capacity": 4, "name": "3F2", "email": os.getenv("GOOGLE_CALENDAR_3F2")},
+    {"capacity": 4, "name": "3F3", "email": os.getenv("GOOGLE_CALENDAR_3F3")},
+    {"capacity": 4, "name": "3F5", "email": os.getenv("GOOGLE_CALENDAR_3F5")},
+    {"capacity": 4, "name": "3F6", "email": os.getenv("GOOGLE_CALENDAR_3F6")},
+    {"capacity": 12, "name": "6F1", "email": os.getenv("GOOGLE_CALENDAR_6F1")},
+    {"capacity": 4, "name": "6F2", "email": os.getenv("GOOGLE_CALENDAR_6F2")},
+    {"capacity": 4, "name": "6F3", "email": os.getenv("GOOGLE_CALENDAR_6F3")},
+    {"capacity": 2, "name": "6F5", "email": os.getenv("GOOGLE_CALENDAR_6F5")},
+    {"capacity": 2, "name": "6F6", "email": os.getenv("GOOGLE_CALENDAR_6F6")},
+    {"capacity": 2, "name": "6F7", "email": os.getenv("GOOGLE_CALENDAR_6F7")},
+    {"capacity": 4, "name": "7F2", "email": os.getenv("GOOGLE_CALENDAR_7F2")},
+    {"capacity": 4, "name": "7F3", "email": os.getenv("GOOGLE_CALENDAR_7F3")},
+    {"capacity": 4, "name": "7F5", "email": os.getenv("GOOGLE_CALENDAR_7F5") },
+    {"capacity": 4, "name": "7F6", "email": os.getenv("GOOGLE_CALENDAR_7F6") }
+]
+
+def get_credentials():
+    creds = None
+    if os.path.exists(TOKEN_FILENAME):
+        creds = Credentials.from_authorized_user_file(TOKEN_FILENAME, SCOPES)
+    
+    # 크리덴셜이 없거나 유효하지 않으면 사용자 로그인 플로우를 실행합니다.
+    if not creds or not creds.valid:
+        if creds and creds.expired and creds.refresh_token:
+            creds.refresh(Request())
+        else:
+            flow = InstalledAppFlow.from_client_secrets_file(
+                CREDENTIALS_FILENAME, SCOPES)
+            creds = flow.run_local_server(port=0)
+        
+        # 다음 실행을 위해 토큰을 저장합니다.
+        with open(TOKEN_FILENAME, 'w') as token:
+            token.write(creds.to_json())
+    return creds
+
+
+def check_room_free_slots_over_1h(date_str=None):
+    # 회의실 정보
+
+    output_lines = []
+    
+    # --- 인증 및 서비스 빌드 ---
+    try:
+        creds = get_credentials()
+        service = build('calendar', 'v3', credentials=creds)
+    except Exception as e:
+        print(f"認証エラー: {e}")
+        return
+
+    # --- 조회 시간 범위 설정 ---
+    # --- 조회 시간 범위 설정 ---
+    tz = pytz.timezone(TIMEZONE)
+    now = datetime.datetime.now(tz)
+
+    if date_str:
+        target_date = datetime.datetime.strptime(date_str, "%Y-%m-%d").date()
+        start_time = tz.localize(datetime.datetime.combine(target_date, datetime.time(10, 0)))
+        end_time = tz.localize(datetime.datetime.combine(target_date, datetime.time(19, 0)))
+        output_lines.append(f"照会対象: {start_time.strftime('%Y/%m/%d')}")
+    else:
+        today = now.date()
+        start_hour = max(10, now.hour)
+        start_time = tz.localize(datetime.datetime.combine(today, datetime.time(start_hour, 0)))
+        end_time = tz.localize(datetime.datetime.combine(today, datetime.time(19, 0)))
+        output_lines.append(f"照会対象: {start_time.strftime('%Y/%m/%d')}")
+
+    if start_time >= end_time:
+        output_lines.append("照会可能な時間がありません")
+        return "\n".join(output_lines)
+
+    # --- FreeBusy API로 모든 회의실의 바쁜 시간 한 번에 조회 ---
+    body = {
+        "timeMin": start_time.isoformat(),
+        "timeMax": end_time.isoformat(),
+        "timeZone": TIMEZONE,
+        "items": [{"id": room["email"]} for room in rooms]
+    }
+
+    try:
+        freebusy_result = service.freebusy().query(body=body).execute()
+        calendars_busy_info = freebusy_result.get('calendars', {})
+    except Exception as e:
+        return f"FreeBusy API 失敗: {e}"
+        
+    output_lines.append("-" * 30)
+
+    # --- 각 회의실별로 빈 시간 계산 및 출력 ---
+    for room in rooms:
+        room_email = room['email']
+        busy_info = calendars_busy_info.get(room_email, {})
+
+        if busy_info.get('errors'):
+            reason = busy_info['errors'][0].get('reason', 'unknown')
+            output_lines.append(f"[{room['name']}] カレンダー照会失敗: {reason}")
+            continue
+
+        busy_slots = busy_info.get('busy', [])
+        
+        cursor = start_time
+        free_slots = []
+
+        for busy in busy_slots:
+            busy_start = datetime.datetime.fromisoformat(busy['start'])
+            busy_end = datetime.datetime.fromisoformat(busy['end'])
+
+            if busy_start > cursor:
+                free_slots.append({'start': cursor, 'end': busy_start})
+            
+            if busy_end > cursor:
+                cursor = busy_end
+
+        if cursor < end_time:
+            free_slots.append({'start': cursor, 'end': end_time})
+
+        long_free_slots = [
+            slot for slot in free_slots
+            if (slot['end'] - slot['start']) >= datetime.timedelta(hours=1)
+        ]
+
+        if long_free_slots:
+            output_lines.append(f"[{room['name']}] (定員 {room['capacity']}名)")
+            for slot in long_free_slots:
+                output_lines.append(f"  {slot['start'].strftime('%H:%M')} ~ {slot['end'].strftime('%H:%M')}")
+    
+    return "\n".join(output_lines)
+
+
+def search_calendar(calendar_data , order: str) -> str:
+    genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+    
+    model = genai.GenerativeModel("gemini-1.5-pro")
+
+    message = f"""以下が空き室の空いている時間情報です。\n {calendar_data} このデータを見て"""
+
+    
+    prompt = "{message}{order}".format(message=message, order=order)   
+    token_info = model.count_tokens(prompt)
+    print(f"Prompt Token Count: {token_info.total_tokens}")
+    
+    response = model.generate_content(prompt)
+    return response.text if response.text else "生成失敗"
+
+
+def calendar_main(input_string: str):
+    try:
+        parts = input_string.split()
+        if len(parts) < 3 or parts[0].lower() != '会議室':
+            return
+
+        # 날짜 파싱 및 포맷팅
+        date_input = parts[1]
+        formatted_date= ""
+        if date_input == "すぐ":
+            formatted_date = ""
+        elif len(date_input) != 8 or not date_input.isdigit():
+            print("日付のフォーマットが正しくありません YYYYMMDD にしてください。")
+            return
+        else:
+            formatted_date = f"{date_input[:4]}-{date_input[4:6]}-{date_input[6:]}"
+
+        # 인원수 파싱
+        capacity_input = parts[2]
+        if not capacity_input.isdigit():
+            print("人数は数字のみ入力してください。")
+            return
+        min_capacity = int(capacity_input)    
+        
+
+        # 4번째 인자가 있으면 취득
+        optional_arg = ""
+        if len(parts) > 3:
+            optional_arg = " ".join(parts[3:])
+            print(f"追加オプション: '{optional_arg}'")
+        
+        
+        
+        
+    except Exception as e:
+        print(f"エラー発生: {e}")
+        return
+    
+    free_rooms = check_room_free_slots_over_1h(formatted_date)
+    if free_rooms:
+        response = search_calendar(free_rooms, f"{formatted_date}に{min_capacity}名が入れる会議室を教えてください{optional_arg}。回答はシンプルに[会議室名] 定員 空き時間のフォーマットでしてください")
+        return response
+    else:
+        return
+
+
+
+