from django.shortcuts import render
from django.http import JsonResponse
import requests
import json

def index(request):
    return render(request, 'chat/index.html')

def send_message(request):
    if request.method == 'POST':
        user_message = request.POST.get('message')
        
        api_url = 'http://192.168.1.36:11434/api/generate'  # Updated URL with the correct port
        headers = {
            'Authorization': 'Bearer your_api_key_here',  # Include your API key if needed
            'Content-Type': 'application/json'
        }
        payload = {
            'prompt': user_message,
            'model': 'mistral:latest'  # Include the model
        }

        try:
            response = requests.post(api_url, json=payload, headers=headers)
            response.raise_for_status()  # This will raise an HTTPError for bad responses
            print(f'Response Content: {response.content}')

            # Handle the response correctly by splitting lines and parsing each JSON object
            response_lines = response.content.decode('utf-8').splitlines()
            combined_response = ""
            for line in response_lines:
                json_obj = json.loads(line)
                if 'response' in json_obj:
                    combined_response += json_obj['response'] + " "
                if json_obj.get('done', False):
                    break

            return JsonResponse({'response': combined_response.strip()})
        except requests.exceptions.HTTPError as errh:
            print(f'HTTP Error: {errh}')
            print(f'Response Content: {response.content}')
            return JsonResponse({'error': f'HTTP Error: {errh}', 'details': response.content.decode('utf-8')}, status=400)
        except requests.exceptions.ConnectionError as errc:
            print(f'Error Connecting: {errc}')
            return JsonResponse({'error': f'Error Connecting: {errc}'}, status=500)
        except requests.exceptions.Timeout as errt:
            print(f'Timeout Error: {errt}')
            return JsonResponse({'error': f'Timeout Error: {errt}'}, status=500)
        except requests.exceptions.RequestException as err:
            print(f'Request Error: {err}')
            print(f'Response Content: {response.content}')
            return JsonResponse({'error': f'Request Error: {err}', 'details': response.content.decode('utf-8')}, status=500)