Upload 4 files

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 IMG_1358.jpg filter=lfs diff=lfs merge=lfs -text
+src/IMG_1358.jpg filter=lfs diff=lfs merge=lfs -text

src/streamlit_app.py

@@ -1,40 +1,449 @@
-import altair as alt
-import numpy as np
-import pandas as pd
-import streamlit as st
-
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+import streamlit as st
+from datetime import datetime, timedelta
+import pytz
+from google_auth_oauthlib.flow import Flow
+from google.oauth2.credentials import Credentials
+from googleapiclient.discovery import build
+from googleapiclient.errors import HttpError
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+import json
+import re
+import ast
+import base64
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+
+st.set_page_config(page_title="ChronoCall-Q", page_icon="🗓️")
+
+st.markdown("""
+    <style>
+    /* เปลี่ยนกรอบหลักให้เป็นสีม่วง */
+    div[data-baseweb="input"] > div {
+        border: 2px solid #a020f0 !important;
+        border-radius: 6px;
+        padding: 2px;
+        box-shadow: none !important;
+    }
+
+    /* ตอน focus แล้ว */
+    div[data-baseweb="input"] > div:focus-within {
+        border: 2px solid #a020f0 !important;
+        box-shadow: 0 0 0 2px rgba(160, 32, 240, 0.3) !important;
+    }
+
+    /* input ด้านในไม่ให้แสดงเงาสีแดงเลย */
+    input {
+        border: none !important;
+        outline: none !important;
+        box-shadow: none !important;
+    }
+
+    /* ลบพวกกรอบแดงที่แอบซ่อนอยู่ */
+    .css-1cpxqw2, .css-1d391kg, .css-1y4p8pa {
+        border: 2px solid #a020f0 !important;
+        box-shadow: none !important;
+    }
+
+    /* force กล่อง input ให้ไม่ใช้สีแดงแม้จะ error */
+    div:has(input:focus) {
+        border-color: #a020f0 !important;
+    }
+
+    /* เปลี่ยนสีข้อความหัวข้อเป็นม่วงเมื่อ hover */
+    details:hover > summary {
+        color: #9D00FF !important;
+        font-weight: bold;
+    }
+
+    /* เปลี่ยนสี caret ^ (ไอคอนสามเหลี่ยม) ตอน hover */
+    details:hover summary::-webkit-details-marker {
+        color: #9D00FF !important;
+    }
+    </style>
+""", unsafe_allow_html=True)
+
+
+# --- Model ---
+
+def convert_to_dict(text):
+    match = re.search(r"<tool_call>\n(.*?)\n</tool_call>", text)
+
+    if match:
+        tool_dict_str = match.group(1)
+        try:
+            result = ast.literal_eval(tool_dict_str)
+            return(result)
+        except Exception as e:
+            return({})
+    else:
+        return({})
+
+# --- Model ---
+
+model_name_or_path = "TechitoTamani/Qwen3-0.6B_FinetuneWithMyData-Merged"
+
+with open('tools.json', 'r', encoding='utf-8') as f:
+    TOOLS = json.load(f)
+
+@st.cache_resource
+def load_model_and_tokenizer():
+    tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
+    model = AutoModelForCausalLM.from_pretrained(
+        model_name_or_path,
+        torch_dtype=torch.bfloat16,
+        device_map="auto",
+        trust_remote_code=True,
+    )
+    return tokenizer, model
+
+tokenizer, model = load_model_and_tokenizer()
+
+def get_model_answer(messages):
+    text = tokenizer.apply_chat_template(
+        messages,
+        tokenize=False,
+        add_generation_prompt=True,
+        tools=TOOLS,
+        enable_thinking=False
+    )
+    inputs = tokenizer(text, return_tensors="pt").to(model.device)
+    with torch.no_grad():
+        outputs = model.generate(**inputs, max_new_tokens=128)
+    output_text = tokenizer.batch_decode(outputs)[0][len(text):]
+    return output_text
+
+# --- Calendar ---
+
+SCOPES = ['https://www.googleapis.com/auth/calendar']
+
+with open("style.css") as f:
+    st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
+
+def get_base64_image(image_path):
+    with open(image_path, "rb") as img_file:
+        b64_data = base64.b64encode(img_file.read()).decode()
+    return b64_data
+
+def create_flow():
+    return Flow.from_client_config(
+        {
+            "web": {
+                "client_id": st.secrets["client_id"],
+                "client_secret": st.secrets["client_secret"],
+                "auth_uri": "https://accounts.google.com/o/oauth2/auth",
+                "token_uri": "https://oauth2.googleapis.com/token",
+                "redirect_uris": [st.secrets["redirect_uri"]],
+            }
+        },
+        scopes=SCOPES,
+        redirect_uri=st.secrets["redirect_uri"]
+    )
+
+def generate_auth_url(flow):
+    auth_url, _ = flow.authorization_url(prompt='consent', access_type='offline', include_granted_scopes='true')
+    return auth_url
+
+def create_service(creds):
+    return build("calendar", "v3", credentials=creds)
+
+# --- Adding ---
+
+def extract_time_or_dash(iso_time: str) -> str:
+    try:
+        dt = datetime.fromisoformat(iso_time)
+
+        if "T" in iso_time:
+            return dt.strftime("%H:%M")
+        else:
+            return "-"
+    except ValueError:
+        return "-"
+
+def handle_calendar_action(service, action_data):
+    name = action_data.get("name")
+    args = action_data.get("arguments", {})
+    
+    if name == "add_event_date":
+        date_str = args["date"]
+        time_str = args["time"]
+        title = args["title"]
+
+        dt_start = datetime.strptime(f"{date_str} {time_str}", "%Y-%m-%d %H:%M")
+        dt_end = dt_start + timedelta(hours=1)
+
+        event = {
+            'summary': title,
+            'start': {
+                'dateTime': dt_start.isoformat(),
+                'timeZone': 'Asia/Bangkok',
+            },
+            'end': {
+                'dateTime': dt_end.isoformat(),
+                'timeZone': 'Asia/Bangkok',
+            },
+        }
+
+        created = service.events().insert(calendarId='primary', body=event).execute()
+        st.success(f"✅ เพิ่มกิจกรรม: {title} [ดูใน Calendar]({created.get('htmlLink')})")
+
+    elif name == "delete_event_date":
+        date_str = args["date"]
+        title = args["title"]
+        events = get_events_by_date_and_title(service, date_str, title)
+        if events:
+            for event in events:
+                service.events().delete(calendarId='primary', eventId=event['id']).execute()
+            st.success(f"🗑 ลบนัด: {title} ในวันที่ {date_str} แล้ว")
+        else:
+            st.warning(f"ไม่พบนัด: {title} ในวันที่ {date_str}")
+
+    elif name == "update_event":
+        date_str = args["date"]
+        new_time_str = args["time"]
+        title = args["title"]
+        events = get_events_by_date_and_title(service, date_str, title)
+
+        if events:
+            for event in events:
+                new_start = datetime.strptime(f"{date_str} {new_time_str}", "%Y-%m-%d %H:%M")
+                new_end = new_start + timedelta(hours=1)
+                event["start"]["dateTime"] = new_start.isoformat()
+                event["end"]["dateTime"] = new_end.isoformat()
+                service.events().update(calendarId='primary', eventId=event["id"], body=event).execute()
+            st.success(f"✏️ เปลี่ยนเวลานัด: {title} เป็น {new_time_str}")
+        else:
+            st.warning(f"ไม่พบนัด: {title} ในวันที่ {date_str}")
+
+    elif name == "view_event_date":
+        date_str = args["date"]
+        events = get_events_by_date(service, date_str)
+        if not events:
+            st.info("ไม่มีนัดในวันนี้")
+        else:
+            st.write(f"📅 นัดทั้งหมดในวันที่ {date_str}:")
+            for e in events:
+                date = e['start'].get('dateTime', e['start'].get('date'))
+                time = extract_time_or_dash(date)
+                st.write(f"- {e['summary']} เวลา: {time}")
+
+def get_events_by_date(service, date_str):
+    date_obj = datetime.strptime(date_str, "%Y-%m-%d")
+    start = date_obj.strftime("%Y-%m-%dT00:00:00Z")
+    end = (date_obj + timedelta(days=1)).strftime("%Y-%m-%dT00:00:00Z")
+    
+    # # DEBUG
+    # st.write(f"Fetching events for date: {date_str}")
+    # st.write(f"timeMin = {start}, timeMax = {end}")
+
+    try:
+        events_result = service.events().list(
+            calendarId='primary',
+            timeMin=start,
+            timeMax=end,
+            singleEvents=True,
+            orderBy='startTime'
+        ).execute()
+        return events_result.get('items', [])
+    
+    except HttpError as error:
+        st.error(f"❌ Google Calendar API error: {error}")
+        return []
+
+    except Exception as e:
+        st.error(f"❌ Unknown error: {e}")
+        return []
+
+def get_events_by_date_and_title(service, date_str, title):
+    events = get_events_by_date(service, date_str)
+    return [e for e in events if e.get("summary") == title]
+
+# --- Email ---
+
+sender = st.secrets["email"]["sender"]
+receiver = st.secrets["email"]["receiver"]
+password = st.secrets["email"]["password"]
+
+def send_email(user_email):
+    subject = "New Beta Tester - ChronoCall-Q"
+    body = f"""\
+    มีผู้สนใจเข้าร่วม Beta Test
+    อีเมล: {user_email}
+    """
+
+    msg = MIMEMultipart()
+    msg["From"] = sender
+    msg["To"] = receiver
+    msg["Subject"] = subject
+    msg.attach(MIMEText(body, "plain"))
+
+    try:
+        with smtplib.SMTP_SSL("smtp.gmail.com", 465) as server:
+            server.login(sender, password)
+            server.sendmail(sender, receiver, msg.as_string())
+        return True
+    except Exception as e:
+        st.error(f"เกิดข้อผิดพลาด: {e}")
+        return False
+
+# --- Streamlit ---
+
+def main():
+
+    # ดึง query params
+    params = st.query_params
+    code = params.get("code", None)
+
+    # ถ้ายังไม่ได้ login
+    if "credentials" not in st.session_state:
+        if code:
+            flow = create_flow()
+            try:
+                flow.fetch_token(code=code)
+                creds = flow.credentials
+                st.session_state["credentials"] = {
+                    "token": creds.token,
+                    "refresh_token": creds.refresh_token,
+                    "token_uri": creds.token_uri,
+                    "client_id": creds.client_id,
+                    "client_secret": creds.client_secret,
+                    "scopes": creds.scopes
+                }
+                st.success("🎉 ล็อกอินสำเร็จ! พร้อมใช้งานแล้ว")
+                st.rerun()
+            except Exception as e:
+                st.error(f"เกิดข้อผิดพลาด: {e}")
+                return
+        else:
+            flow = create_flow()
+            auth_url = generate_auth_url(flow)
+
+            # --- Title + Login ---
+
+            st.markdown(f"""
+                <!-- ชื่อ -->
+                <div class="fade-in-title custom-title">ChronoCall-Q</div>
+
+                <!-- ปุ่ม ล่าช้า -->
+                <div class="fade-in-button login-button">
+                    <a href="{auth_url}" target="_blank" rel="noopener noreferrer">
+                        <button>Login with Google</button>
+                    </a>
+                </div>
+            """, unsafe_allow_html=True)
+
+            # --- Emali sender ---
+
+            st.markdown("""
+            <div class="fade-in-up">
+                <p>ถ้า <strong>Login ไม่ได้ใช่ไหม?</strong><br>ไม่เป็นไร! ส่งคำขอเข้าร่วม Beta Test มาให้เราก่อนได้เลย</p>
+            </div>
+            """, unsafe_allow_html=True)
+
+            email = st.text_input("กรอกอีเมลของคุณที่นี่")
+
+            if st.button("ส่งคำขอเข้าร่วม Beta Test"):
+                if email:
+                    if send_email(email):
+                        st.success("ส่งคำขอเรียบร้อยแล้ว!")
+                    else:
+                        st.error("เกิดข้อผิดพลาดในการส่ง กรุณาลองใหม่อีกครั้ง")
+                else:
+                    st.warning("กรุณากรอกอีเมลก่อนกดส่ง")
+
+            # --- pic ---
+
+            img_b64 = get_base64_image("IMG_1358.jpg")
+
+            st.markdown(f'''
+                <img src="data:image/jpeg;base64,{img_b64}" class="fade-in-image">
+            ''', unsafe_allow_html=True)
+
+            st.stop()
+
+    # Logged in
+    creds = Credentials(**st.session_state["credentials"])
+    service = create_service(creds)
+
+    st.title("ChronoCall-Q")
+    st.caption("วิธีใช้งาน")
+
+    with st.expander("การเพิ่มเหตุการณ์ (Add Event)"):
+        st.markdown("""
+    เพื่อเพิ่มเหตุการณ์ใหม่ ต้องระบุ:
+    - **ชื่อเหตุการณ์**
+    - **วันที่**
+    - **เวลา**
+
+    > **ตัวอย่าง:** เพิ่มนัดประชุมพรุ่งนี้ตอนสิบโมง
+    """)
+
+    with st.expander("การลบเหตุการณ์ (Delete Event)"):
+        st.markdown("""
+    เพื่อทำการลบเหตุการณ์ ต้องระบุ:
+    - **ชื��อเหตุการณ์**
+    - **วันที่**
+
+    > **ตัวอย่าง:** เพิ่มนัดประชุมพรุ่งนี้ตอนสิบโมง
+    """)
+
+    with st.expander("การอัปเดตเหตุการณ์ (Update Event)"):
+        st.markdown("""
+    เพื่อเปลี่ยนเวลาเหตุการณ์ในวันเดียวกัน ต้องระบุ:
+    - **ชื่อเหตุการณ์**
+    - **วันที่**
+    - **เวลาใหม่**
+
+    > **ตัวอย่าง:** เปลี่ยนนัดประชุมพรุ่งนี้เป็นสิบเอ็ดโมง
+    """)
+
+    with st.expander("การดูเหตุการณ์ (View Events)"):
+        st.markdown("""
+    เพื่อดูรายการเหตุการณ์ในวันใดวันหนึ่ง ต้องระบุ:
+    - **วันที่ที่ต้องการดู**
+
+    > **ตัวอย่าง:** พรุ่งนี้มีนัดอะไรบ้าง
+    """)
+
+    if "user_input" not in st.session_state:
+        st.session_state.user_input = ""
+
+    messages = [
+        {"role": "system", "content": "You are Qwen, created by Alibaba Cloud. You are a helpful assistant.\n\nCurrent Date: 2025-06-07.\n\nCurrent Day: Saturday."},
+    ]
+
+
+    user_input = st.text_input("พิมพ์คำสั่งที่นี่ แล้วกด Enter หรือกดปุ่มยืนยัน", value=st.session_state.user_input, key="input")
+
+    submit_button = st.button("ยืนยัน")
+
+    if (user_input and user_input != st.session_state.user_input) or submit_button:
+
+        bangkok_tz = pytz.timezone("Asia/Bangkok")
+        now = datetime.now(bangkok_tz)
+        current_date = now.strftime("%Y-%m-%d")
+        current_day = now.strftime("%A")
+
+        messages = [
+            {
+                "role": "system",
+                "content": f"You are Qwen, created by Alibaba Cloud. You are a helpful assistant.\n\nCurrent Date: {current_date}.\n\nCurrent Day: {current_day}."
+            },
+            {
+                "role": "user",
+                "content": user_input
+            }
+        ]
+
+        with st.spinner("โมเดลกำลังคิด..."):
+            response = get_model_answer(messages)
+
+        st.success(f"Chrono: {response}")
+        func_call_dict = convert_to_dict(response)
+        handle_calendar_action(service, func_call_dict)
+
+        st.session_state.user_input = ""
+
+if __name__ == "__main__":
+    main()

src/style.css

@@ -0,0 +1,86 @@
+@keyframes fadeInTitle {
+    from { opacity: 0; transform: translateY(-20px); }
+    to { opacity: 1; transform: translateY(0); }
+}
+
+@keyframes fadeInButton {
+    from { opacity: 0; transform: translateY(20px); }
+    to { opacity: 1; transform: translateY(0); }
+}
+
+.fade-in-title {
+    animation: fadeInTitle 1s ease-out;
+}
+
+.fade-in-button {
+    animation: fadeInButton 1s ease-out;
+    animation-delay: 1s;
+    animation-fill-mode: both;
+}
+
+.custom-title {
+    font-size: 60px;
+    font-weight: 800;
+    color: #ffffff;
+    text-align: center;
+    margin-bottom: 30px;
+}
+
+.login-button {
+    display: flex;
+    justify-content: center;
+}
+
+.login-button a button {
+    background-color: #663399;
+    border: none;
+    color: white;
+    padding: 10px 24px;
+    font-size: 16px;
+    border-radius: 8px;
+    cursor: pointer;
+    font-weight: bold;
+    box-shadow: 0 4px 10px rgba(106, 13, 173, 0.5);
+    transition: background-color 0.3s ease, box-shadow 0.3s ease;
+}
+
+.login-button a button:hover {
+    background-color: #7b29e1;
+    box-shadow: 0 6px 14px rgba(123, 41, 225, 0.7);
+}
+
+@keyframes fadeIn {
+  from {opacity: 0;}
+  to {opacity: 1;}
+}
+
+.fade-in-image {
+  animation: fadeIn 1.5s ease-in forwards;
+  width: 100%;
+  max-width: 600px;
+  border-radius: 15px;
+  box-shadow: 0 4px 15px rgba(0,0,0,0.2);
+  display: block;
+  margin: 40px auto 40px auto;  /* เพิ่ม margin-top 40px */
+}
+
+@keyframes fadeInUp {
+    from {
+        opacity: 0;
+        transform: translateY(20px);
+    }
+    to {
+        opacity: 1;
+        transform: translateY(0);
+    }
+}
+.fade-in-up {
+    opacity: 0;
+    animation: fadeInUp 1s ease forwards;
+    animation-delay: 1.5s;
+    margin-top: 40px;
+    margin-bottom: 20px;
+    color: #ffffff;
+    font-size: 16px;
+    text-align: center;
+}

src/tools.json

@@ -0,0 +1,105 @@
+[
+  {
+    "type": "function",
+    "function": {
+      "name": "add_event_date",
+      "description": "Add event by title and date.",
+      "parameters": {
+        "type": "object",
+        "properties": {
+          "title": {
+            "type": "string",
+            "description": "The title of event ."
+          },
+          "date": {
+            "type": "string",
+            "description": "The date to add events for, in the format \"Year-Month-Day\". The selected date must be within one year from Current Date."
+          },
+          "time": {
+            "type": "string",
+            "description": "Optional. The time to add events at, in the format \"HH:MM\". Leave blank if not applicable."
+          }
+        },
+        "required": [
+          "title",
+          "date"
+        ]
+      }
+    }
+  },
+  {
+    "type": "function",
+    "function": {
+      "name": "delete_event_date",
+      "description": "Delete event by title and date.",
+      "parameters": {
+        "type": "object",
+        "properties": {
+          "title": {
+            "type": "string",
+            "description": "The title of event that want to delete."
+          },
+          "date": {
+            "type": "string",
+            "description": "The date to delete events for, in the format \"Year-Month-Day\". The selected date must be within one year from Current Date."
+          }
+        },
+        "required": [
+          "title",
+          "date"
+        ]
+      }
+    }
+  },
+  {
+    "type": "function",
+    "function": {
+      "name": "update_event",
+      "description": "update event by title and date.",
+      "parameters": {
+        "type": "object",
+        "properties": {
+          "title": {
+            "type": "string",
+            "description": "The title of event that want to update."
+          },
+          "date": {
+            "type": "string",
+            "description": "The date to update events for, in the format \"Year-Month-Day\". The selected date must be within one year from Current Date."
+          },
+          "time": {
+            "type": "string",
+            "description": "Optional. The time to update events at, in the format \"HH:MM\". Leave blank if not applicable."
+          }
+        },
+        "required": [
+          "title",
+          "date"
+        ]
+      }
+    }
+  },
+  {
+    "type": "function",
+    "function": {
+      "name": "view_event_date",
+      "description": "view event by date.",
+      "parameters": {
+        "type": "object",
+        "properties": {
+          "date": {
+            "type": "string",
+            "description": "The date to view events for, in the format \"Year-Month-Day\". The selected date must be within one year from Current Date."
+          },
+          "time": {
+            "type": "string",
+            "description": "Optional. The time to view events at, in the format \"HH:MM\". Leave blank if not applicable."
+          }
+        },
+        "required": [
+          "date"
+        ]
+      }
+    }
+  }
+]