Lumaforge / lumaforge /ollama_client.py

Commit History

feat: implement premium cinematic typography layouts, revert ControlNet, and remove token
f47d70f

sujithputta commited on

fix: Improve Ollama client logging and use llama3.2:1b model - Add debug logging on OllamaClient initialization - Update model to llama3.2:1b (faster, lighter weight) - Improve connection check logging with clear status messages - Graceful fallback to heuristic prompts when Ollama unavailable Ollama is running on localhost:11434 with multiple models available: - llama3.2:1b (1.2B, fastest) - llama3:latest (8B, best quality) - mistral:latest (7.2B, balanced) - deepseek-r1:7b (7.6B, best reasoning)
5680fa0

sujithputta commited on

fix: Improve Ollama fallback handling and add better debug logging - Reduce Ollama timeout from 35s to 10s for faster fallback - Add TimeoutError handling alongside URLError - Improve debug messages for model loading stages - Enable mock mode by default (can be toggled via API) - Add more granular logging for MPS pipeline initialization This fixes the hanging issue when Ollama is not running by quickly falling back to mock mode with better error reporting.
52db35c

sujithputta commited on

v1.1: Add 16 specialized categories, colorization, face restoration, and advanced image editing Features: - Add category_prompts.py with 16 specialized categories (110+ templates) * Creative Art, Characters, Landscapes, Architecture, Vehicles, Products, Marketing * Food, Fashion, Gaming, Animals, Events, Business, Education * Each category includes 5-8 optimized subcategories with specialized prompt templates - Implement colorization endpoint (/api/colorize) with 5 styles * Vibrant, Warm, Cool, Vintage, Sepia color grading styles * Adaptive color enhancement with saturation and contrast control - Implement face restoration endpoint (/api/face-restoration) with 4 intensity levels * Low, Medium, High, Ultra intensity levels * Denoising, sharpening, contrast enhancement, color vibrancy boost - Update pipeline.py with new methods * colorize() method supporting 5 color grading styles * restore_face() method with intensity-based enhancement - Update app.py with new request models * ColorizeRequest model with style parameter * FaceRestorationRequest model with intensity parameter * Add 2 new endpoints: /api/colorize and /api/face-restoration - Update ollama_client.py expand_prompt() method * Add category and subcategory parameter support * Integrate category-specific prompt enhancements * Maintain backward compatibility Backend: - All endpoints tested and verified - Proper error handling and base64 encoding/decoding - Rate limiting applied to new endpoints - Production-ready implementation
0914032

sujithputta commited on

Deploy LumaForge AuraGen backend API to Hugging Face
6836a23

sujithputta commited on