transpordi_artiklid/query_client.py

#!/usr/bin/env python3
"""
Päringukliendi näide Weaviate'i baasile
"""

import sys
import os
import json
from datetime import datetime

# Lisa src kaust Pythoni teele
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))

from src.query_engine import QueryEngine

def print_result(result: dict):
    """Prindi tulemus ilusalt"""
    print("\n" + "="*80)
    print(f"PÄRING: {result.get('query', '')}")
    print(f"TÜÜP: {result.get('search_type', '')} | LEITI: {result.get('articles_found', 0)} artiklit")
    print(f"USALDUSVÄÄRSUS: {result.get('confidence', 0):.2%}")
    print("="*80)
    
    # Vastus
    print("\n📝 VASTUS:")
    print(result.get('answer', ''))
    
    # Allikad
    sources = result.get('sources', [])
    if sources:
        print(f"\n📚 ALLIKAD ({len(sources)}):")
        for i, source in enumerate(sources, 1):
            print(f"\n{i}. {source.get('title', '')}")
            print(f"   Autorid: {', '.join(source.get('authors', []))}")
            print(f"   Aasta: {source.get('year', '')} | Žurnaal: {source.get('journal', '')}")
            print(f"   Relevantsus: {source.get('relevance_score', 5)}/10")
            print(f"   Skoor: {source.get('score', 0):.3f}")
            if source.get('summary'):
                print(f"   Kokkuvõte: {source.get('summary')}")
    
    print("\n" + "="*80)
    print(f"Kuupäev: {result.get('timestamp', '')}")

def interactive_mode():
    """Interaktiivne režiim"""
    print("🤖 TRANSFORDIARTIKLITE PÄRINGUMOOTOR")
    print("="*60)
    print("Kasuta järgmisi käske:")
    print("  ? - Abi")
    print("  q - Välju")
    print("  t [päring] - Tehniline vastus")
    print("  l [päring] - Lühike vastus")
    print("  d [päring] - Detailne vastus (vaikimisi)")
    print("  s [päring] - Semantiline otsing")
    print("  k [päring] - Võtmesõnade otsing")
    print("  h [päring] - Hübriidotsing (vaikimisi)")
    print("="*60)
    
    engine = QueryEngine()
    
    while True:
        try:
            user_input = input("\n> ").strip()
            
            if not user_input:
                continue
            
            if user_input.lower() in ['q', 'quit', 'exit']:
                print("Head aega!")
                break
            
            if user_input == '?':
                print("""
Käsud:
  t [päring] - Tehniline vastus (spetsialistidele)
  l [päring] - Lühike vastus (kokkuvõte)
  d [päring] - Detailne vastus (põhjalik analüüs)
  s [päring] - Semantiline otsing (mõttesisu)
  k [päring] - Võtmesõnade otsing (tekstipõhine)
  h [päring] - Hübriidotsing (mõlemad)
  
Näited:
  t kuidas parandada teede ohutust
  l elektriautod linnaliikluses
  d jalgrataste ja autode jagatud teed
                """)
                continue
            
            # Parse command
            parts = user_input.split(' ', 1)
            if len(parts) < 2:
                print("❌ Vigane käsk. Kasuta: [käsk] [päring]")
                continue
            
            command, query = parts[0].lower(), parts[1]
            
            # Determine parameters
            if command == 't':
                context_type = "technical"
                search_type = "hybrid"
            elif command == 'l':
                context_type = "concise"
                search_type = "hybrid"
            elif command == 'd':
                context_type = "detailed"
                search_type = "hybrid"
            elif command == 's':
                context_type = "detailed"
                search_type = "semantic"
            elif command == 'k':
                context_type = "detailed"
                search_type = "keyword"
            elif command == 'h':
                context_type = "detailed"
                search_type = "hybrid"
            else:
                print(f"❌ Tundmatu käsk: {command}")
                continue
            
            print(f"🔍 Otsin: '{query}'...")
            result = engine.ask(query, search_type=search_type, context_type=context_type)
            print_result(result)
            
        except KeyboardInterrupt:
            print("\n\nVäljun...")
            break
        except Exception as e:
            print(f"❌ Viga: {str(e)}")
    
    engine.close()

def batch_mode(queries_file: str):
    """Pakettrežiim - päringud failist"""
    try:
        with open(queries_file, 'r', encoding='utf-8') as f:
            queries = [line.strip() for line in f if line.strip()]
    except FileNotFoundError:
        print(f"❌ Faili {queries_file} ei leitud")
        return
    
    engine = QueryEngine()
    results = []
    
    print(f"⏳ Töötlen {len(queries)} päringut...")
    
    for i, query in enumerate(queries, 1):
        print(f"  {i}/{len(queries)}: '{query}'")
        
        try:
            result = engine.ask(query)
            results.append(result)
            
            # Salvesta iga tulemus eraldi faili
            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
            filename = f"tmp/query_result_{timestamp}_{i}.json"
            
            with open(filename, 'w', encoding='utf-8') as f:
                json.dump(result, f, ensure_ascii=False, indent=2)
            
            print(f"    ✓ Salvestatud {filename}")
            
        except Exception as e:
            print(f"    ✗ Viga: {str(e)}")
            results.append({
                "query": query,
                "error": str(e),
                "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
            })
    
    engine.close()
    
    # Salvesta kõik tulemused ühte faili
    summary_file = f"tmp/query_summary_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json"
    with open(summary_file, 'w', encoding='utf-8') as f:
        json.dump(results, f, ensure_ascii=False, indent=2)
    
    print(f"\n✅ Valmis! Tulemused salvestatud faili: {summary_file}")

def single_query(query: str, search_type: str = "hybrid", context_type: str = "detailed"):
    """Üksik päring"""
    engine = QueryEngine()
    
    print(f"🔍 Päring: {query}")
    result = engine.ask(query, search_type=search_type, context_type=context_type)
    
    print_result(result)
    engine.close()
    
    # Salvesta tulemus
    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
    filename = f"tmp/query_{timestamp}.json"
    
    with open(filename, 'w', encoding='utf-8') as f:
        json.dump(result, f, ensure_ascii=False, indent=2)
    
    print(f"📄 Tulemus salvestatud faili: {filename}")

if __name__ == "__main__":
    import argparse
    
    parser = argparse.ArgumentParser(description="Transpordiartiklite päringumootor")
    parser.add_argument("--query", "-q", help="Üksik päring")
    parser.add_argument("--file", "-f", help="Päringute fail (üks päring real)")
    parser.add_argument("--type", "-t", choices=["semantic", "keyword", "hybrid"], 
                       default="hybrid", help="Otsingu tüüp")
    parser.add_argument("--context", "-c", choices=["detailed", "concise", "technical"], 
                       default="detailed", help="Vastuse tüüp")
    parser.add_argument("--interactive", "-i", action="store_true", 
                       help="Interaktiivne režiim")
    
    args = parser.parse_args()
    
    if args.interactive:
        interactive_mode()
    elif args.file:
        batch_mode(args.file)
    elif args.query:
        single_query(args.query, args.type, args.context)
    else:
        # Vaikimisi interaktiivne režiim
        interactive_mode()