import os
import requests
import json

def get_ollama_host():
    host = input("Introduce adresa pentru OLLAMA_HOST (ex: https://example.ngrok-free.app): ")
    os.environ['OLLAMA_HOST'] = host
    return host

def chat_with_llama(host):
    print("\nChat started. Type 'exit' to quit.")
    
    while True:
        message = input("\nYou: ")
        if message.lower() == 'exit':
            break
            
        try:
            response = requests.post(
                f"{host}/api/chat",
                json={
                    "model": "llama3.3",
                    "messages": [{"role": "user", "content": message}],
                    "stream": False
                }
            )
            response.raise_for_status()
            print(f"\nAssistant: {response.json()['message']['content']}")
        except Exception as e:
            print(f"\nError: {str(e)}")

if __name__ == "__main__":
    host = get_ollama_host()
    print(f"\nOLLAMA_HOST setat la: {host}")
    chat_with_llama(host)