AI + автоматизация

#!/usr/bin/env python3
"""ai_process.py — универсальный скрипт для обработки текста через Ollama."""

import requests
import json
import sys

OLLAMA = "http://localhost:11434/api/chat"
MODEL = "qwen3.5:4b"

def ai_process(text: str, instruction: str = "Ответь кратко") -> str:
    """Отправляет текст модели и возвращает ответ."""
    
    response = requests.post(OLLAMA, json={
        "model": MODEL,
        "messages": [
            {"role": "system", "content": instruction},
            {"role": "user", "content": text}
        ],
        "stream": False
    })
    return response.json()["message"]["content"]

def ai_process_json(text: str, json_schema: dict) -> dict:
    """Отправляет текст и получает структурированный JSON."""
    
    response = requests.post(OLLAMA, json={
        "model": MODEL,
        "messages": [{"role": "user", "content": text}],
        "format": json_schema,
        "stream": False
    })
    return json.loads(response.json()["message"]["content"])


# Примеры использования
if __name__ == "__main__":
    # Простой вызов
    result = ai_process(
        "Напиши краткое саммари: 'Компания выпустила новую версию продукта...'",
        "Ты — ассистент. Составляй краткие саммари (2-3 предложения)."
    )
    print(f"📝 {result}")
    
    # JSON-вывод
    schema = {
        "type": "object",
        "properties": {
            "summary": {"type": "string"},
            "sentiment": {"type": "string", "enum": ["positive", "negative", "neutral"]},
            "keywords": {"type": "array", "items": {"type": "string"}}
        }
    }
    data = ai_process_json(
        "Продукт отличный, но интерфейс немного запутанный",
        schema
    )
    print(f"📊 {json.dumps(data, ensure_ascii=False, indent=2)}")

#!/usr/bin/env python3
"""telegram_ai_bot.py — Telegram-бот на локальной модели."""

import requests
import logging
from http.server import HTTPServer, BaseHTTPRequestHandler

OLLAMA = "http://localhost:11434/api/chat"
MODEL = "qwen3.5:4b"

class AIBotHandler(BaseHTTPRequestHandler):
    """Простой HTTP-сервер, принимающий запросы и отвечающий через AI."""

    def do_POST(self):
        content_length = int(self.headers['Content-Length'])
        body = self.rfile.read(content_length).decode()

        response = requests.post(OLLAMA, json={
            "model": MODEL,
            "messages": [
                {"role": "system", "content": "Ты — полезный ассистент. Отвечай кратко."},
                {"role": "user", "content": body}
            ],
            "stream": False
        })

        answer = response.json()["message"]["content"]

        self.send_response(200)
        self.send_header('Content-type', 'application/json')
        self.end_headers()
        self.wfile.write(f'{{"response": "{answer}"}}'.encode())

if __name__ == "__main__":
    server = HTTPServer(("localhost", 8888), AIBotHandler)
    print("🤖 AI Bot запущен на http://localhost:8888")
    server.serve_forever()