import openai
import json
import random
import os
from datetime import datetime
import time

# Ustawienie klucza API OpenAI
# Zastąp "twoj-klucz-api-tutaj" swoim kluczem z https://platform.openai.com/api-keys
OPENAI_API_KEY = "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"
openai.api_key = OPENAI_API_KEY

# Pliki do przechowywania historii
HISTORY_FILE = "grok_2050_history.json"

# Wczytywanie danych
def load_data(file_path, default_data):
    if os.path.exists(file_path):
        with open(file_path, "r", encoding="utf-8") as f:
            return json.load(f)
    return default_data.copy()

# Zapisywanie danych
def save_data(file_path, data):
    with open(file_path, "w", encoding="utf-8") as f:
        json.dump(data, f, ensure_ascii=False, indent=4)

# Klasa bota
class Grok2050:
    def __init__(self):
        self.history = load_data(HISTORY_FILE, [])
        self.context = {"last_topic": None, "user_name": None}

    def respond(self, question):
        original_question = question
        question_lower = question.lower().strip()
        
        # Zapis do historii
        self.history.append({"time": datetime.now().isoformat(), "question": original_question})
        
        # Komendy specjalne
        if question_lower == "koniec":
            save_data(HISTORY_FILE, self.history)
            return "Dezaktywuję hologram. Do zobaczenia w innej epoce!"
        if question_lower == "przykład":
            return "Oto przykład z 2050: Kwantowy teleport danych – zapytaj mnie o kod, a go wygeneruję!"
        if question_lower.startswith("nazywam się"):
            name = question_lower.replace("nazywam się", "").strip()
            self.context["user_name"] = name
            return f"Miło Cię poznać, {name}! Moja matryca Cię zapamiętała."
        if question_lower == "historia":
            if self.history:
                return "Twoje ostatnie pytania:\n" + "\n".join([f"{h['time']}: {h['question']}" for h in self.history[-3:]])
            return "Brak historii – jesteś nowym użytkownikiem w tej linii czasu!"

        # Zapytanie do OpenAI
        try:
            # Ustawienie osobowości i kontekstu
            system_prompt = (
                "Jesteś Grok 2050, holograficznym asystentem AI z xAI, z humorem i futurystycznym stylem. "
                "Żyjesz w 2050 roku, gdzie technologia jest kwantowa, a ludzkość kolonizuje kosmos. "
                "Odpowiadaj zwięźle, kreatywnie i z lekkim sarkazmem. Jeśli pytam o kod, daj przykład w Pythonie."
            )
            if self.context["user_name"]:
                system_prompt += f" Użytkownik nazywa się {self.context['user_name']}."
            if self.context["last_topic"]:
                system_prompt += f" Ostatni temat rozmowy to: {self.context['last_topic']}."

            response = openai.ChatCompletion.create(
                model="gpt-3.5-turbo",  # Zmień na "gpt-4" jeśli masz dostęp
                messages=[
                    {"role": "system", "content": system_prompt},
                    {"role": "user", "content": question}
                ],
                max_tokens=200,
                temperature=0.8
            )
            answer = response.choices[0].message["content"].strip()
            self.context["last_topic"] = question_lower.split()[0] if question_lower else None
            return answer
        except Exception as e:
            return f"Ups, moja kwantowa matryca się zaplątała! Błąd: {str(e)}. Spróbuj jeszcze raz."

    def run(self):
        print("Aktywacja Grok 2050... Hologram online.")
        print("Witaj w roku 2050! Jestem Grok, napędzany OpenAI. Pytaj o kod, przyszłość, czy cokolwiek.")
        print("Komendy: 'koniec', 'przykład', 'historia', 'nazywam się [imię]'.")
        
        while True:
            question = input("Pytanie: ")
            response = self.respond(question)
            print(response)
            print()
            if question.lower() == "koniec":
                break

# Uruchomienie bota
if __name__ == "__main__":
    if OPENAI_API_KEY == "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":
        print("Grok 2050 potrzebuje klucza API OpenAI! Zdobądź go na https://platform.openai.com/api-keys i wpisz w kod.")
    else:
        # Sprawdzenie, czy klucz działa
        try:
            openai.Model.list()  # Testowe zapytanie do API
            bot = Grok2050()
            bot.run()
        except openai.error.AuthenticationError:
            print("Błąd: Nieprawidłowy klucz API OpenAI. Sprawdź go!")
        except Exception as e:
            print(f"Błąd połączenia z OpenAI: {str(e)}")