Update src/chatbot.py

src/chatbot.py

@@ -18,38 +18,58 @@ from dotenv import load_dotenv
 class TDKChatbot:
     """TDK Sözlük RAG Chatbot."""
 
-    def __init__(self, api_key=None, vector_store_path="./data/vector_store"):
+    def __init__(self, api_key=None, vector_store_path=None):
         """
         Args:
             api_key: Gemini API anahtarı
             vector_store_path: Vector store dosya yolu
         """
+        from data_downloader import DataDownloader  # İÇERİDE IMPORT
+        
         # Environment variables yükle
         load_dotenv()
-
+    
         # API key kontrolü
         self.api_key = api_key or os.getenv('GEMINI_API_KEY')
         if not self.api_key:
-            raise ValueError("GEMINI_API_KEY bulunamadı! .env dosyasını kontrol edin.")
-
-        print("TDK Chatbot başlatılıyor...")
-
+            raise ValueError("GEMINI_API_KEY bulunamadı!")
+    
+        print("=" * 70)
+        print("TDK CHATBOT BAŞLATILIYOR")
+        print("=" * 70)
+    
+        # DATASET'TEN DOSYALARI İNDİR
+        downloader = DataDownloader()
+        if not downloader.check_files():
+            print("\nVector store bulunamadı, dataset'ten indiriliyor...")
+            if not downloader.download_and_setup():
+                raise ValueError("Dosyalar indirilemedi!")
+        else:
+            print("Vector store dosyaları mevcut\n")
+    
         # Gemini'yi yapılandır
         genai.configure(api_key=self.api_key)
-
-        # Gemini modelini seç (2.0 Flash - hızlı ve güçlü)
         self.model = genai.GenerativeModel('gemini-2.0-flash-exp')
-        print("Gemini 2.0 Flash modeli yüklendi")
-
+        print("Gemini 2.0 Flash hazır")
+    
         # Embedding modelini yükle
         print("Embedding modeli yükleniyor...")
         self.embedder = EmbeddingModel()
-
+    
         # Vector store'u yükle
         print("Vector store yükleniyor...")
         self.vector_store = FAISSVectorStore()
+        
+        if vector_store_path is None:
+            vector_store_path = "./data/vector_store"
+        
         if not self.vector_store.load(vector_store_path):
             raise ValueError("Vector store yüklenemedi!")
+    
+        print("\n" + "=" * 70)
+        print("CHATBOT HAZIR!")
+        print("=" * 70)
+        print()
 
         print("Chatbot hazır!\n")