BF / app.py
SamiKoen's picture
Upload app.py
8b6b7a8 verified
raw
history blame
27.2 kB
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)