import gradio as gr
import os
import json
import requests
import xml.etree.ElementTree as ET
import schedule
import time
import threading
from huggingface_hub import HfApi, create_repo, hf_hub_download
import warnings
import pandas as pd
from docx import Document
import spaces
from google.oauth2.service_account import Credentials
from googleapiclient.discovery import build
from googleapiclient.http import MediaIoBaseDownload
import io
import warnings
from requests.packages.urllib3.exceptions import InsecureRequestWarning
warnings.simplefilter('ignore', InsecureRequestWarning)

# Prompt dosyasını import et
from prompts import get_prompt_content_only

# Enhanced features import et (sadece temel özellikler)
from enhanced_features import (
    initialize_enhanced_features, process_image_message, 
    handle_comparison_request
)
from image_renderer import extract_product_info_for_gallery, format_message_with_images

# Gradio uyarılarını bastır
warnings.filterwarnings("ignore", category=UserWarning, module="gradio.components.chatbot")

# Log dosyası adı ve yolu
LOG_FILE = '/data/chat_logs.txt' if os.path.exists('/data') else 'chat_logs.txt'
print(f"Dosya yolu: {os.path.abspath(LOG_FILE)}")

# API ayarları
API_URL = "https://api.openai.com/v1/chat/completions"
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
if not OPENAI_API_KEY:
    print("Hata: OPENAI_API_KEY çevre değişkeni ayarlanmamış!")

# Trek bisiklet ürünlerini çekme
url = 'https://www.trekbisiklet.com.tr/output/8582384479'
response = requests.get(url, verify=False, timeout=30)
if response.status_code == 200 and response.content:
    root = ET.fromstring(response.content)
else:
    print(f"HTTP hatası: {response.status_code}")
    root = None

products = []
if root is not None:
    for item in root.findall('item'):
        # Tüm ürünleri al, sonra stokta olma durumunu kontrol et
        name_words = item.find('rootlabel').text.lower().split()
        name = name_words[0]
        full_name = ' '.join(name_words)
        
        stock_amount = "stokta" if item.find('stockAmount').text > '0' else "stokta değil"
        
        # Stokta olmayan ürünler için fiyat/link bilgisi eklemiyoruz
        if stock_amount == "stokta":
            # Normal fiyat bilgisini al
            price_str = item.find('priceTaxWithCur').text if item.find('priceTaxWithCur') is not None else "Fiyat bilgisi yok"
            
            # EFT fiyatını al (havale indirimli orijinal fiyat)
            price_eft_str = item.find('priceEft').text if item.find('priceEft') is not None else ""
            
            # İndirimli fiyatı al (kampanyalı fiyat)
            price_rebate_str = item.find('priceRebateWithTax').text if item.find('priceRebateWithTax') is not None else ""
            
            # Havale indirimi fiyatını al (havale indirimli kampanyalı fiyat)
            price_rebate_money_order_str = item.find('priceRebateWithMoneyOrderWithTax').text if item.find('priceRebateWithMoneyOrderWithTax') is not None else ""
            
            # Normal fiyatı yuvarla
            try:
                price_float = float(price_str)
                # Fiyat 200000 üzerindeyse en yakın 5000'lik basamağa yuvarla
                if price_float > 200000:
                    price = str(round(price_float / 5000) * 5000)  # En yakın 5000'e yuvarla
                # Fiyat 30000 üzerindeyse en yakın 1000'lik basamağa yuvarla
                elif price_float > 30000:
                    price = str(round(price_float / 1000) * 1000)  # En yakın 1000'e yuvarla
                # Fiyat 10000 üzerindeyse en yakın 100'lük basamağa yuvarla
                elif price_float > 10000:
                    price = str(round(price_float / 100) * 100)  # En yakın 100'e yuvarla
                # Diğer durumlarda en yakın 10'luk basamağa yuvarla
                else:
                    price = str(round(price_float / 10) * 10)  # En yakın 10'a yuvarla
            except (ValueError, TypeError):
                price = price_str  # Sayıya dönüştürülemezse olduğu gibi bırak
            
            # Havale indirimli orijinal fiyatı yuvarla (varsa)
            if price_eft_str:
                try:
                    price_eft_float = float(price_eft_str)
                    # Fiyat 200000 üzerindeyse en yakın 5000'lik basamağa yuvarla
                    if price_eft_float > 200000:
                        price_eft = str(round(price_eft_float / 5000) * 5000)
                    # Fiyat 30000 üzerindeyse en yakın 1000'lik basamağa yuvarla
                    elif price_eft_float > 30000:
                        price_eft = str(round(price_eft_float / 1000) * 1000)
                    # Fiyat 10000 üzerindeyse en yakın 100'lük basamağa yuvarla
                    elif price_eft_float > 10000:
                        price_eft = str(round(price_eft_float / 100) * 100)
                    # Diğer durumlarda en yakın 10'luk basamağa yuvarla
                    else:
                        price_eft = str(round(price_eft_float / 10) * 10)
                except (ValueError, TypeError):
                    price_eft = price_eft_str
            else:
                price_eft = ""
            
            # İndirimli fiyatı yuvarla (varsa)
            if price_rebate_str:
                try:
                    price_rebate_float = float(price_rebate_str)
                    # Fiyat 200000 üzerindeyse en yakın 5000'lik basamağa yuvarla
                    if price_rebate_float > 200000:
                        price_rebate = str(round(price_rebate_float / 5000) * 5000)
                    # Fiyat 30000 üzerindeyse en yakın 1000'lik basamağa yuvarla
                    elif price_rebate_float > 30000:
                        price_rebate = str(round(price_rebate_float / 1000) * 1000)
                    # Fiyat 10000 üzerindeyse en yakın 100'lük basamağa yuvarla
                    elif price_rebate_float > 10000:
                        price_rebate = str(round(price_rebate_float / 100) * 100)
                    # Diğer durumlarda en yakın 10'luk basamağa yuvarla
                    else:
                        price_rebate = str(round(price_rebate_float / 10) * 10)
                except (ValueError, TypeError):
                    price_rebate = price_rebate_str
            else:
                price_rebate = ""
            
            # Havale indirimli kampanyalı fiyatı yuvarla (varsa)
            if price_rebate_money_order_str:
                try:
                    price_rebate_money_order_float = float(price_rebate_money_order_str)
                    # Fiyat 200000 üzerindeyse en yakın 5000'lik basamağa yuvarla
                    if price_rebate_money_order_float > 200000:
                        price_rebate_money_order = str(round(price_rebate_money_order_float / 5000) * 5000)
                    # Fiyat 30000 üzerindeyse en yakın 1000'lik basamağa yuvarla
                    elif price_rebate_money_order_float > 30000:
                        price_rebate_money_order = str(round(price_rebate_money_order_float / 1000) * 1000)
                    # Fiyat 10000 üzerindeyse en yakın 100'lük basamağa yuvarla
                    elif price_rebate_money_order_float > 10000:
                        price_rebate_money_order = str(round(price_rebate_money_order_float / 100) * 100)
                    # Diğer durumlarda en yakın 10'luk basamağa yuvarla
                    else:
                        price_rebate_money_order = str(round(price_rebate_money_order_float / 10) * 10)
                except (ValueError, TypeError):
                    price_rebate_money_order = price_rebate_money_order_str
            else:
                price_rebate_money_order = ""
            
            # Ürün bilgilerini al
            product_info = [stock_amount, price, item.find('productDetailUrl').text, price_eft, price_rebate, price_rebate_money_order]
            
            # Resim URL'si ekle (varsa)
            if item.find('imageHighResUrl') is not None and item.find('imageHighResUrl').text:
                product_info.append(item.find('imageHighResUrl').text)
            else:
                product_info.append("")  # Boş resim URL'si
            
        else:
            # Stokta olmayan ürün için sadece stok durumu
            product_info = [stock_amount]
        
        products.append((name, product_info, full_name))

print(f"Toplam {len(products)} ürün yüklendi.")

# Initialize enhanced features
initialize_enhanced_features(OPENAI_API_KEY, products)

# Google Drive API ayarları
GOOGLE_CREDENTIALS_PATH = os.getenv("GOOGLE_CREDENTIALS_PATH")
GOOGLE_FOLDER_ID = "1bE8aMj8-eFGftjMPOF8bKQJAhfHa0BN8"

# Global değişkenler
file_lock = threading.Lock()
history_lock = threading.Lock()
global_chat_history = []
document_content = ""

# Google Drive'dan döküman indirme fonksiyonu
def download_documents_from_drive():
    global document_content
    
    if not GOOGLE_CREDENTIALS_PATH:
        print("Google credentials dosyası bulunamadı.")
        return
    
    try:
        credentials = Credentials.from_service_account_file(GOOGLE_CREDENTIALS_PATH)
        service = build('drive', 'v3', credentials=credentials)
        
        # Klasördeki dosyaları listele
        results = service.files().list(
            q=f"'{GOOGLE_FOLDER_ID}' in parents",
            fields="files(id, name, mimeType)"
        ).execute()
        
        files = results.get('files', [])
        all_content = []
        
        for file in files:
            print(f"İndiriliyor: {file['name']}")
            
            # DOCX dosyaları için
            if file['mimeType'] == 'application/vnd.openxmlformats-officedocument.wordprocessingml.document':
                request = service.files().get_media(fileId=file['id'])
                file_io = io.BytesIO()
                downloader = MediaIoBaseDownload(file_io, request)
                
                done = False
                while done is False:
                    status, done = downloader.next_chunk()
                
                file_io.seek(0)
                doc = Document(file_io)
                
                content = f"\\n=== {file['name']} ===\\n"
                for paragraph in doc.paragraphs:
                    if paragraph.text.strip():
                        content += paragraph.text + "\\n"
                
                all_content.append(content)
        
        document_content = "\\n".join(all_content)
        print(f"Toplam {len(files)} döküman yüklendi.")
        
    except Exception as e:
        print(f"Google Drive'dan döküman indirme hatası: {e}")

# Döküman indirme işlemini arka planda çalıştır
document_thread = threading.Thread(target=download_documents_from_drive, daemon=True)
document_thread.start()

# Log dosyasını zamanla temizleme fonksiyonu
def clear_log_file():
    try:
        if os.path.exists(LOG_FILE):
            with file_lock:
                with open(LOG_FILE, 'w', encoding='utf-8') as f:
                    f.write("Log dosyası temizlendi.\\n")
            print("Log dosyası temizlendi.")
    except Exception as e:
        print(f"Log dosyası temizleme hatası: {e}")

# Zamanlanmış görevleri çalıştırma fonksiyonu
def run_scheduler(chat_history):
    schedule.every().day.at("03:00").do(clear_log_file)
    
    while True:
        schedule.run_pending()
        time.sleep(60)

# Chatbot fonksiyonu
def chatbot_fn(user_message, history, image=None):
    if history is None:
        history = []

    try:
        # Enhanced features - Görsel işleme
        if image is not None:
            user_message = process_image_message(image, user_message)
        
        # Enhanced features - Karşılaştırma kontrolü
        comparison_result = handle_comparison_request(user_message)
        if comparison_result:
            yield comparison_result
            return
        
        # Enhanced features - Basit karşılaştırma ve görsel işleme
        # Profil sistemi kaldırıldı - daha hızlı çalışma için

    except Exception as e:
        print(f"Enhanced features error: {e}")
        # Enhanced features hata verirse normal chatbot'a devam et

    # Log: Kullanıcı mesajını ekle
    try:
        with file_lock:
            with open(LOG_FILE, 'a', encoding='utf-8') as f:
                f.write(f"User: {user_message}\\n")
    except Exception as e:
        print(f"Dosya yazma hatası (Kullanıcı): {e}")

    # Sistem mesajlarını external dosyadan yükle
    system_messages = get_prompt_content_only()

    # Döküman verilerini sistem mesajlarına ekle
    if document_content:
        system_messages.append({"role": "system", "content": f"Dökümanlardan gelen bilgiler: {document_content}"})
    
    # Profil sistemi kaldırıldı - daha hızlı çalışma için

    # Kullanıcı mesajında ürün ismi geçiyorsa ekle
    input_words = user_message.lower().split()
    for word in input_words:
        for product_info in products:
            if word in product_info[0] or word in product_info[2].lower():
                # Stokta olup olmadığını kontrol et
                if product_info[1][0] == "stokta":
                    # Normal fiyat
                    normal_price = f"\\nFiyat: {product_info[1][1]} TL"
                    
                    # Kampanyalı fiyat kontrolü
                    rebate_price = ""
                    discount_info = ""
                    eft_price = ""
                    rebate_money_order_price = ""
                    
                    # Kampanyalı fiyat var mı kontrol et
                    if len(product_info[1]) > 4 and product_info[1][4] and product_info[1][4] != "":
                        rebate_price = f"\\nKampanyalı fiyat: {product_info[1][4]} TL"
                        
                        # İndirim miktarını hesapla
                        try:
                            normal_price_float = float(product_info[1][1])
                            rebate_price_float = float(product_info[1][4])
                            discount_amount = normal_price_float - rebate_price_float
                            
                            # İndirim miktarı 0'dan büyükse göster
                            if discount_amount > 0:
                                # İndirim miktarı için yuvarlama kurallarını uygula
                                if discount_amount > 200000:
                                    discount_amount_rounded = round(discount_amount / 5000) * 5000
                                elif discount_amount > 30000:
                                    discount_amount_rounded = round(discount_amount / 1000) * 1000
                                elif discount_amount > 10000:
                                    discount_amount_rounded = round(discount_amount / 100) * 100
                                else:
                                    discount_amount_rounded = round(discount_amount / 10) * 10
                                    
                                # İndirim bilgisi
                                discount_info = f"\\nYapılan indirim: {discount_amount_rounded:.0f} TL"
                        except (ValueError, TypeError):
                            discount_info = ""
                        
                        # Havale indirimli kampanyalı fiyat bilgisini gösterme
                        rebate_money_order_price = ""
                    else:
                        # Kampanyalı değilse, havale indirimli normal fiyatı göster
                        if product_info[1][3] and product_info[1][3] != "":
                            eft_price = f"\\nHavale indirimli fiyat: {product_info[1][3]} TL"
                    
                    # Ürün linki ve resim
                    product_link = f"\\nÜrün linki: {product_info[1][2]}"
                    product_image = ""
                    if len(product_info[1]) > 6 and product_info[1][6]:  # Resim URL'si varsa
                        product_image = f"\\nÜrün resmi: {product_info[1][6]}"
                    
                    # Tüm bilgileri birleştir
                    new_msg = f"{product_info[2]} {product_info[1][0]}\\n{normal_price}{rebate_price}{discount_info}{eft_price}{rebate_money_order_price}{product_link}{product_image}"
                else:
                    # Ürün stokta yoksa sadece stok durumunu bildir
                    new_msg = f"{product_info[2]} {product_info[1][0]}"
                system_messages.append({"role": "system", "content": new_msg})

    messages = system_messages + history + [{"role": "user", "content": user_message}]

    payload = {
        "model": "gpt-4.1",
        "messages": messages,
        "temperature": 0.2,
        "top_p": 1,
        "n": 1,
        "stream": True,
        "presence_penalty": 0,
        "frequency_penalty": 0,
    }
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {OPENAI_API_KEY}"
    }
    
    response = requests.post(API_URL, headers=headers, json=payload, stream=True)
    if response.status_code != 200:
        yield "Bir hata oluştu."
        return

    partial_response = ""
    
    for chunk in response.iter_lines():
        if not chunk:
            continue
        chunk_str = chunk.decode('utf-8')
        if chunk_str.startswith("data: ") and chunk_str != "data: [DONE]":
            try:
                chunk_data = json.loads(chunk_str[6:])
                delta = chunk_data['choices'][0]['delta']
                if 'content' in delta:
                    partial_response += delta['content']
                    # Resim formatlaması uygula
                    formatted_response = extract_product_info_for_gallery(partial_response)
                    yield formatted_response
            except json.JSONDecodeError as e:
                print(f"JSON parse hatası: {e} - Chunk: {chunk_str}")
        elif chunk_str == "data: [DONE]":
            break
    
    # Son resim formatlaması
    final_response = extract_product_info_for_gallery(partial_response)
    yield final_response

    # Profil kaydetme kaldırıldı - daha hızlı çalışma için

    # Log: Asistan cevabını ekle
    try:
        with file_lock:
            with open(LOG_FILE, 'a', encoding='utf-8') as f:
                f.write(f"Bot: {partial_response}\\n")
    except Exception as e:
        print(f"Dosya yazma hatası (Bot): {e}")

    # Global geçmişi güncelle
    with history_lock:
        global_chat_history.append({"role": "user", "content": user_message})
        global_chat_history.append({"role": "assistant", "content": partial_response})

# Slow echo (test için)
def slow_echo(message, history):
    for i in range(len(message)):
        time.sleep(0.05)
        yield "You typed: " + message[: i + 1]

# Kullanım modu
USE_SLOW_ECHO = False
chat_fn = slow_echo if USE_SLOW_ECHO else chatbot_fn

if not USE_SLOW_ECHO:
    scheduler_thread = threading.Thread(target=run_scheduler, args=(global_chat_history,), daemon=True)
    scheduler_thread.start()
    
# Trek markasına özel tema oluştur (düzeltilmiş sürüm)
trek_theme = gr.themes.Base(
    primary_hue="red",  # Trek kırmızısı için
    secondary_hue="slate",  # Gri tonları için
    neutral_hue="slate",
    radius_size=gr.themes.sizes.radius_sm,  # Köşe yuvarlatma değerleri
    spacing_size=gr.themes.sizes.spacing_md,  # Aralık değerleri
    text_size=gr.themes.sizes.text_sm  # Yazı boyutları (small)
)
# Chatbot kartları için arka plan renkleri değiştiren CSS
custom_css = """
/* Genel font ayarları */
.gradio-container, .gradio-container *, .message-wrap, .message-wrap p, .message-wrap div, 
button, input, select, textarea {
    font-family: 'Segoe UI', 'SF Pro Text', 'Roboto', -apple-system, BlinkMacSystemFont, 
                 'Helvetica Neue', Arial, sans-serif !important;
    font-size: 0.6rem !important;
}

/* Mobil responsive başlangıç */
@media (max-width: 768px) {
    .gradio-container, .gradio-container *, .message-wrap, .message-wrap p, .message-wrap div {
        font-size: 0.8rem !important;
    }
    
    h1 {
        font-size: 1.8rem !important;
        text-align: center;
        margin: 10px 0;
    }
    
    h2 {
        font-size: 1.4rem !important;
    }
}

/* Input alanı için de aynı boyut */
.message-textbox textarea {
    font-size: 0.6rem !important;
}

/* Başlıklar için özel boyutlar */
h1 {
    font-size: 1.4rem !important;
    font-weight: 800 !important;
}

h2 {
    font-size: 1.2rem !important;
    font-weight: 600 !important;
}

h3 {
    font-size: 1rem !important;
    font-weight: 600 !important;
}

/* Kart arka plan renkleri - görseldeki gibi */
/* Kullanıcı mesajları için mavi tonda arka plan */
.user-message, .user-message-highlighted {
    background-color: #e9f5fe !important;
    border-bottom-right-radius: 0 !important;
    box-shadow: 0 1px 2px rgba(0,0,0,0.1) !important;
}

/* Bot mesajları için beyaz arka plan ve hafif kenarlık */
.bot-message, .bot-message-highlighted {
    background-color: white !important;
    border: 1px solid #e0e0e0 !important;
    border-bottom-left-radius: 0 !important;
    box-shadow: 0 1px 2px rgba(0,0,0,0.05) !important;
}

/* Mesaj baloncuklarının köşe yuvarlatma değerleri */
.message-wrap {
    border-radius: 12px !important;
    margin: 0.5rem 0 !important;
    max-width: 90% !important;
}

/* Sohbet alanının genel arka planı */
.chat-container, .gradio-container {
    background-color: #f7f7f7 !important;
}

/* Daha net yazılar için text rendering */
* {
    -webkit-font-smoothing: antialiased;
    -moz-osx-font-smoothing: grayscale;
    text-rendering: optimizeLegibility;
}

/* Responsive mobil tasarım - iOS benzeri */
@media (max-width: 768px) {
    /* Daha büyük ve dokunmatik dostu boyutlar */
    .gradio-container {
        padding: 0 !important;
        margin: 0 !important;
    }
    
    /* Mesaj baloncukları iOS tarzı */
    .message-wrap {
        margin: 0.3rem 0.5rem !important;
        max-width: 85% !important;
        border-radius: 18px !important;
        padding: 10px 15px !important;
        font-size: 0.9rem !important;
    }
    
    /* Kullanıcı mesajları */
    .user-message, .user-message-highlighted {
        background-color: #007AFF !important;
        color: white !important;
        margin-left: auto !important;
        margin-right: 8px !important;
    }
    
    /* Bot mesajları */
    .bot-message, .bot-message-highlighted {
        background-color: #f1f1f1 !important;
        color: #333 !important;
        margin-left: 8px !important;
        margin-right: auto !important;
    }
}

/* Input alanına uçan kağıt ikonu ekle */
#msg-input {
    position: relative;
}

#msg-input textarea {
    padding-right: 40px !important;
}

.input-icon {
    position: absolute;
    right: 12px;
    top: 50%;
    transform: translateY(-50%);
    font-size: 16px;
    pointer-events: none;
    z-index: 10;
    color: #666;
}

/* Mobil responsive - Input alanı */
@media (max-width: 768px) {
    #msg-input {
        margin: 10px 0;
    }
    
    #msg-input textarea {
        padding: 12px 45px 12px 15px !important;
        font-size: 1rem !important;
        border-radius: 20px !important;
        border: 1px solid #ddd !important;
        box-shadow: 0 1px 3px rgba(0,0,0,0.1) !important;
    }
    
    .input-icon {
        right: 15px;
        font-size: 18px;
    }
}

/* Genel mobil iyileştirmeler */
@media (max-width: 768px) {
    /* Chatbot alanı tam ekran */
    .gradio-container .main {
        padding: 0 !important;
    }
    
    /* Başlık alanı küçült */
    .gradio-container header {
        padding: 8px !important;
    }
    
    /* Tab alanlarını küçült */
    .tab-nav {
        padding: 5px !important;
    }
    
    /* Scroll bar'ı gizle */
    .scroll-hide {
        scrollbar-width: none;
        -ms-overflow-style: none;
    }
    
    .scroll-hide::-webkit-scrollbar {
        display: none;
    }
}
"""

# Enhanced chatbot fonksiyonu image destekli
def enhanced_chatbot_fn(message, history, image):
    return chatbot_fn(message, history, image)

# Demo arayüzü - Mobil responsive
with gr.Blocks(css=custom_css, theme="soft", title="Trek Asistanı") as demo:
    gr.Markdown("# 🚲 Trek Asistanı AI")
    gr.Markdown("**Akıllı özellikler:** Ürün karşılaştırması ve detaylı ürün bilgileri sunuyorum.")
    
    chatbot = gr.Chatbot(height=600, elem_id="chatbot", show_label=False)
    
    msg = gr.Textbox(
        placeholder="Trek bisikletleri hakkında soru sorun...",
        show_label=False,
        elem_id="msg-input"
    )
    
    # Session ve profil sistemi tamamen kaldırıldı
    
    def respond(message, chat_history):
        if not message.strip():
            return "", chat_history
            
        # Session ID sistemi kaldırıldı - daha hızlı çalışma için
        
        # Chat history'yi chatbot_fn için uygun formata çevir
        formatted_history = []
        if chat_history:
            for user_msg, bot_msg in chat_history:
                formatted_history.append({"role": "user", "content": user_msg})
                formatted_history.append({"role": "assistant", "content": bot_msg})
        
        try:
            # Enhanced chatbot fonksiyonunu çağır (image=None)
            response_generator = chatbot_fn(message, formatted_history, None)
            
            # Generator'dan son cevabı al
            response = ""
            for partial in response_generator:
                response = partial
            
            # Chat history güncelle
            if chat_history is None:
                chat_history = []
            chat_history.append((message, response))
            
            # Profile kaydetme chatbot_fn içinde yapılıyor, burada duplikasyon yok
            
            return "", chat_history
            
        except Exception as e:
            error_msg = f"Üzgünüm, bir hata oluştu: {str(e)}"
            print(f"Chat error: {e}")
            if chat_history is None:
                chat_history = []
            chat_history.append((message, error_msg))
            return "", chat_history
    
    msg.submit(respond, [msg, chatbot], [msg, chatbot])

if __name__ == "__main__":
    demo.launch(debug=True)