t_s_m.py

# -*- coding: utf-8 -*-
"""T S M.ipynb

Automatically generated by Colab.

Original file is located at
    https://colab.research.google.com/drive/1aa7j80PEKtklLchunk-SKWhzhwCj-PMr
"""

# ==============================
# INSTALL DEPENDENCIES
# ==============================
!pip install -q openai gradio pandas

# ==============================
# IMPORT LIBRARIES
# ==============================
from openai import OpenAI
import os
import gradio as gr
import pandas as pd
from datetime import datetime

# ==============================
# ADD YOUR GROQ API KEY HERE
# ==============================
os.environ["GROQ_API_KEY"] = "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"

# ==============================
# SETUP GROQ CLIENT
# ==============================
client = OpenAI(
    api_key=os.environ.get("GROQ_API_KEY"),
    base_url="https://api.groq.com/openai/v1",
)

# ==============================
# GLOBAL STORAGE
# ==============================
users = {}       # {username: password}
bookings = []    # list of booking dicts
notifications = []  # list of messages to users
ADMIN_CREDENTIALS = {"admin": "admin123"}  # default admin

# ==============================
# SYSTEM PROMPT (AI TRAVEL ASSISTANT)
# ==============================
system_prompt = """
You are TravelGenZ AI, a smart travel assistant.
You give:
1. Distance between departure and destination
2. Airplane vs Bus comparison (time, cost, comfort)
3. Popular attractions & best hotels
4. Estimated cost & travel tips
5. Friendly, GenZ tone with emojis
6. Keep structured output
"""

# ==============================
# USER FUNCTIONS
# ==============================
def register_user(name, password):
    if name in users:
        return "❌ User already exists"
    users[name] = password
    return f"✅ Registered successfully! Welcome, {name}"

def login_user(name, password):
    if name not in users:
        return "❌ User not found. Please register."
    if users[name] != password:
        return "❌ Wrong password."
    return f"✅ Logged in successfully! Welcome back, {name}"

# Travel Chat AI
def travel_chat_ai(departure, destination, travelers, mode, date, name):
    prompt = f"{system_prompt}\nUser is {name}. Traveling from {departure} to {destination} on {date} for {travelers} people by {mode}. Provide distance, travel comparison, popular places and hotels, estimated cost."
    response = client.responses.create(
        model="openai/gpt-oss-20b",
        input=prompt
    )
    return response.output_text

# Make booking
def make_booking(name, departure, destination, travelers, mode, date, mobile):
    booking_id = len(bookings)+1
    booking = {
        "Booking ID": booking_id,
        "Name": name,
        "Mobile": mobile,
        "Departure": departure,
        "Destination": destination,
        "Travelers": travelers,
        "Mode": mode,
        "Date": date,
        "Status": "Pending"
    }
    bookings.append(booking)
    return f"✅ Booking submitted! Your Booking ID is {booking_id}. Admin will review soon."

# Admin functions
def admin_login(name, password):
    if name not in ADMIN_CREDENTIALS or ADMIN_CREDENTIALS[name] != password:
        return "❌ Wrong Admin credentials"
    return "✅ Admin login successful"

def view_bookings(admin_name, admin_password):
    if admin_name not in ADMIN_CREDENTIALS or ADMIN_CREDENTIALS[admin_name] != admin_password:
        return "❌ Wrong Admin credentials"
    if not bookings:
        return "No bookings yet."
    return pd.DataFrame(bookings)

def process_booking(admin_name, admin_password, booking_id, action):
    if admin_name not in ADMIN_CREDENTIALS or ADMIN_CREDENTIALS[admin_name] != admin_password:
        return "❌ Wrong Admin credentials"
    global bookings
    for b in bookings:
        if b["Booking ID"] == int(booking_id):
            if action == "Accept":
                b["Status"] = "Accepted"
                notifications.append(f"✅ Your booking {booking_id} has been accepted by admin!")
                return f"Booking {booking_id} accepted"
            elif action == "Reject":
                b["Status"] = "Rejected"
                notifications.append(f"❌ Your booking {booking_id} has been rejected by admin.")
                return f"Booking {booking_id} rejected"
    return "Booking ID not found"

def get_notifications():
    return "\n".join(notifications) if notifications else "No new messages"

# ==============================
# GRADIO UI
# ==============================
with gr.Blocks(theme=gr.themes.Soft(), title="TravelGenZ AI") as app:

    gr.Markdown("# 🌍✈ TravelGenZ AI Booking System")
    gr.Markdown("### Professional travel booking with AI guidance, hotels & attractions 🎒")

    with gr.Tab("📝 User Login/Register"):
        with gr.Row():
            reg_name = gr.Textbox(label="Name")
            reg_pass = gr.Textbox(label="Password", type="password")
            reg_btn = gr.Button("Register")
        reg_out = gr.Textbox()
        reg_btn.click(register_user, [reg_name, reg_pass], reg_out)

        with gr.Row():
            login_name = gr.Textbox(label="Name")
            login_pass = gr.Textbox(label="Password", type="password")
            login_btn = gr.Button("Login")
        login_out = gr.Textbox()
        login_btn.click(login_user, [login_name, login_pass], login_out)

    with gr.Tab("💬 Travel Assistant"):
        dep_city = gr.Textbox(label="Departure City")
        dest_city = gr.Textbox(label="Destination City")
        travelers = gr.Number(label="Number of Travelers")
        travel_mode = gr.Radio(["Airplane", "Bus"], label="Travel Mode")
        travel_date = gr.Textbox(label="Travel Date (YYYY-MM-DD)")
        user_name = gr.Textbox(label="Your Name")
        mobile = gr.Textbox(label="Mobile Number")

        chat_btn = gr.Button("Get AI Travel Info & Book")
        chat_out = gr.Textbox()

        chat_btn.click(
            lambda d, dst, t, m, date, n, mob: travel_chat_ai(d, dst, t, m, date, n) + "\n\n" + make_booking(n, d, dst, t, m, date, mob),
            [dep_city, dest_city, travelers, travel_mode, travel_date, user_name, mobile],
            chat_out
        )

    with gr.Tab("📨 User Notifications"):
        notif_btn = gr.Button("Check Notifications")
        notif_out = gr.Textbox()
        notif_btn.click(get_notifications, [], notif_out)

    with gr.Tab("👨‍💼 Admin Panel"):
        admin_name = gr.Textbox(label="Admin Name")
        admin_pass = gr.Textbox(label="Password", type="password")
        admin_login_btn = gr.Button("Login")
        admin_login_out = gr.Textbox()
        admin_login_btn.click(admin_login, [admin_name, admin_pass], admin_login_out)

        view_btn = gr.Button("View Bookings")
        bookings_out = gr.Dataframe()
        view_btn.click(view_bookings, [admin_name, admin_pass], bookings_out)

        booking_id = gr.Number(label="Booking ID")
        action = gr.Radio(["Accept", "Reject"], label="Action")
        process_btn = gr.Button("Process Booking")
        process_out = gr.Textbox()
        process_btn.click(process_booking, [admin_name, admin_pass, booking_id, action], process_out)

app.launch(share=True)

# ==============================
# INSTALL DEPENDENCIES
# ==============================
!pip install -q openai gradio pandas

# ==============================
# IMPORT LIBRARIES
# ==============================
from openai import OpenAI
import os
import gradio as gr
import pandas as pd
from datetime import datetime

# ==============================
# SET YOUR GROQ KEY HERE
# ==============================
os.environ["GROQ_API_KEY"] = "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"

# ==============================
# GROQ CLIENT
# ==============================
client = OpenAI(
    api_key=os.environ.get("GROQ_API_KEY"),
    base_url="https://api.groq.com/openai/v1",
)

# ==============================
# GLOBAL STORAGE
# ==============================
users = {}  # username -> password
bookings = []  # all bookings
notifications = {}  # username -> messages
ADMIN_USERNAME = "admin"
ADMIN_PASSWORD = "admin123"

# ==============================
# HELPER FUNCTIONS
# ==============================
def ai_travel_guide(departure, destination, travelers, mode):
    """Get AI-guided info about destination, hotels, attractions, distance & estimated cost"""
    prompt = f"""
    You are a professional travel guide.
    A user wants to travel from {departure} to {destination}.
    Travelers: {travelers}, Mode: {mode}.
    Provide:
    1. Distance in km
    2. Estimated travel time
    3. Estimated cost (simulate realistic)
    4. Best hotels (top 3)
    5. Popular attractions & museums
    Format it with sections and emojis.
    """
    response = client.responses.create(
        model="openai/gpt-oss-20b",
        input=prompt
    )
    return response.output_text

def register_user(username, password):
    if username in users:
        return "❌ Username already exists."
    users[username] = password
    notifications[username] = []
    return "✅ Registration successful! Please login."

def login_user(username, password):
    if username not in users:
        return None, "❌ User not found. Please register."
    if users[username] != password:
        return None, "❌ Incorrect password."
    return username, f"✅ Welcome {username}! Ready to plan your travel journey ✈️🌍"

def make_booking(username, departure, destination, travelers, mode, mobile):
    booking_id = len(bookings) + 1
    time_now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    booking = {
        "Booking ID": booking_id,
        "Username": username,
        "Departure": departure,
        "Destination": destination,
        "Travelers": travelers,
        "Mode": mode,
        "Mobile": mobile,
        "Status": "Pending",
        "Time": time_now
    }
    bookings.append(booking)
    return f"✅ Booking submitted! Your booking ID is {booking_id}. Admin will review it shortly."

def admin_login(username, password):
    if username != ADMIN_USERNAME or password != ADMIN_PASSWORD:
        return None, "❌ Incorrect admin credentials."
    return True, f"✅ Welcome Admin {username}!"

def admin_accept_reject(booking_id, action):
    global bookings
    for b in bookings:
        if b["Booking ID"] == int(booking_id):
            if action == "Accept":
                b["Status"] = "Accepted"
                # Notify user
                msg = f"""
🎉✨ Your Booking is Confirmed! ✨🎉
Hello {b['Username']}! 👋
Great news! Your travel plans with T S M from {b['Departure']} to {b['Destination']} are now confirmed! 🛫🏨🌏
Travelers: {b['Travelers']}, Mode: {b['Mode']}
Your flight & hotel details are all set 📝✈️🏨
Safe travels & happy adventures! 🚀💖🧳
"""
                notifications[b["Username"]].append(msg)
                return f"✅ Booking {booking_id} accepted & notification sent!"
            else:
                b["Status"] = "Rejected"
                notifications[b["Username"]].append(f"❌ Booking {booking_id} has been rejected by admin.")
                return f"✅ Booking {booking_id} rejected."
    return f"❌ Booking ID {booking_id} not found."

# ==============================
# GRADIO UI
# ==============================
with gr.Blocks(theme=gr.themes.Soft(), title="TSM Travel Agency") as app:
    gr.Markdown("# 🌟 TSM Travel Agency AI 🌟")
    gr.Markdown("Welcome to TSM Travel Agency! Plan your journey, explore destinations & book hassle-free ✈️🏨🌍")

    with gr.Tab("📝 User Portal"):
        # Registration/Login
        reg_username = gr.Textbox(label="Username")
        reg_password = gr.Textbox(label="Password", type="password")
        reg_btn = gr.Button("Register")
        login_btn = gr.Button("Login")
        login_msg = gr.Textbox()
        current_user = gr.State()

        reg_btn.click(register_user, [reg_username, reg_password], login_msg)
        login_btn.click(login_user, [reg_username, reg_password], [current_user, login_msg])

        # Travel Planning (after login)
        dep_city = gr.Textbox(label="Departure City")
        dest_city = gr.Textbox(label="Destination City")
        travelers = gr.Number(label="Number of Travelers")
        travel_mode = gr.Radio(["Airplane", "Bus"], label="Travel Mode")
        mobile = gr.Textbox(label="Mobile Number")
        guide_btn = gr.Button("Get Travel Guide & Book")
        guide_output = gr.Textbox()
        booking_output = gr.Textbox()

        def guide_and_book(dep, dest, trav, mode, username, mobile):
            if not username:
                return "❌ Please login first.", ""
            guide_info = ai_travel_guide(dep, dest, trav, mode)
            book_msg = make_booking(username, dep, dest, trav, mode, mobile)
            return guide_info, book_msg

        guide_btn.click(guide_and_book, [dep_city, dest_city, travelers, travel_mode, current_user, mobile], [guide_output, booking_output])

    with gr.Tab("👨‍💼 Admin Panel"):
        admin_user = gr.Textbox(label="Admin Username")
        admin_pass = gr.Textbox(label="Admin Password", type="password")
        admin_login_btn = gr.Button("Login as Admin")
        admin_login_msg = gr.Textbox()
        admin_state = gr.State()

        view_bookings_btn = gr.Button("View All Bookings")
        admin_bookings_output = gr.Dataframe()
        booking_id_input = gr.Number(label="Booking ID")
        accept_btn = gr.Button("Accept Booking")
        reject_btn = gr.Button("Reject Booking")
        admin_action_output = gr.Textbox()

        admin_login_btn.click(admin_login, [admin_user, admin_pass], [admin_state, admin_login_msg])
        view_bookings_btn.click(lambda: pd.DataFrame(bookings), [], admin_bookings_output)
        accept_btn.click(admin_accept_reject, [booking_id_input, gr.State("Accept")], admin_action_output)
        reject_btn.click(admin_accept_reject, [booking_id_input, gr.State("Reject")], admin_action_output)

    with gr.Tab("📬 Notifications"):
        notif_user = gr.Textbox(label="Enter your username to see notifications")
        notif_output = gr.Textbox()
        def show_notifications(u):
            msgs = notifications.get(u, [])
            if not msgs:
                return "No notifications yet."
            return "\n\n".join(msgs)
        gr.Button("Show Notifications").click(show_notifications, notif_user, notif_output)

app.launch(share=True)