Spaces:

andresampa
/

LS-AI-img-gen

Running

App Files Files Community

Andre commited on Feb 10

Commit

6420441

1 Parent(s): 5496988

“Update”

Browse files

Files changed (7) hide show

.DS_Store +0 -0
index_old.php → gallery_content.php +2 -21
gallery_functions.php +28 -0
index.php +6 -188
index_asc.php +8 -0
src/img_gen copy.py +0 -96
src/img_gen.py +2 -2

.DS_Store CHANGED Viewed

Binary files a/.DS_Store and b/.DS_Store differ

index_old.php → gallery_content.php RENAMED Viewed

@@ -1,22 +1,3 @@
-<?php
-// Function to get supported image files
-function getImages($directory) {
-    $allowedTypes = array('jpg', 'jpeg', 'png', 'gif');
-    $files = array();
-    if (is_dir($directory)) {
-        $dirContent = scandir($directory);
-        foreach ($dirContent as $file) {
-            $extension = strtolower(pathinfo($file, PATHINFO_EXTENSION));
-            if (in_array($extension, $allowedTypes)) {
-                $files[] = $file;
-            }
-        }
-    }
-    return $files;
-}
-?>
 <!DOCTYPE html>
 <html lang="en">
 <head>
@@ -132,8 +113,8 @@ function getImages($directory) {
         <div class="gallery">
             <?php
             // Specify your images directory
-            $imageDirectory = 'images/';
-            $images = getImages($imageDirectory);
             foreach ($images as $image) {
                 echo '<div class="gallery-item" onclick="openLightbox(\'' . $imageDirectory . $image . '\')">';

 <!DOCTYPE html>
 <html lang="en">
 <head>
         <div class="gallery">
             <?php
             // Specify your images directory
+            //$imageDirectory = 'images/';
+            //$images = getImages($imageDirectory);
             foreach ($images as $image) {
                 echo '<div class="gallery-item" onclick="openLightbox(\'' . $imageDirectory . $image . '\')">';

gallery_functions.php ADDED Viewed

	@@ -0,0 +1,28 @@

+<?php
+function getImages($directory, $imageOrder = 'desc') {
+    $allowedTypes = array('jpg', 'jpeg', 'png', 'gif');
+    $files = array();
+    if (is_dir($directory)) {
+        $dirContent = scandir($directory);
+        foreach ($dirContent as $file) {
+            $extension = strtolower(pathinfo($file, PATHINFO_EXTENSION));
+            if (in_array($extension, $allowedTypes)) {
+                $files[] = $file;
+            }
+        }
+       // Use strtolower for case-insensitive comparison
+        $order = strtolower($imageOrder); // Normalize the input
+        if ($order == 'desc') {
+            rsort($files);
+        } elseif ($order == 'asc') {
+            sort($files);
+        } // Add other sorting options if needed
+    }
+    return $files;
+}
+?>

index.php CHANGED Viewed

@@ -1,190 +1,8 @@
 <?php
-// Function to get supported image files
-function getImages($directory) {
-    $allowedTypes = array('jpg', 'jpeg', 'png', 'gif');
-    $files = array();
-    if (is_dir($directory)) {
-        $dirContent = scandir($directory);
-        foreach ($dirContent as $file) {
-            $extension = strtolower(pathinfo($file, PATHINFO_EXTENSION));
-            if (in_array($extension, $allowedTypes)) {
-                $files[] = $file;
-            }
-        }
-        // Sort files in reverse order
-        rsort($files);
-    }
-    return $files;
-}
-?>
-<!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Dark Theme Image Gallery</title>
-    <style>
-        body {
-            background-color: #1a1a1a;
-            color: #ffffff;
-            font-family: Arial, sans-serif;
-            margin: 0;
-            padding: 20px;
-        }
-        .gallery-container {
-            max-width: 1200px;
-            margin: 0 auto;
-        }
-        h1 {
-            text-align: center;
-            color: #ffffff;
-            margin-bottom: 30px;
-        }
-        .gallery {
-            display: grid;
-            grid-template-columns: repeat(auto-fill, minmax(250px, 1fr));
-            gap: 20px;
-            padding: 20px;
-        }
-        .gallery-item {
-            position: relative;
-            overflow: hidden;
-            border-radius: 8px;
-            background-color: #2d2d2d;
-            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.3);
-            transition: transform 0.3s ease;
-            cursor: pointer;
-        }
-        .gallery-item:hover {
-            transform: scale(1.02);
-        }
-        .gallery-item img {
-            width: 100%;
-            height: 200px;
-            object-fit: cover;
-            display: block;
-        }
-        .image-caption {
-            padding: 10px;
-            background-color: rgba(0, 0, 0, 0.7);
-            color: #ffffff;
-            font-size: 0.9em;
-            text-align: center;
-        }
-        /* Lightbox styles */
-        .lightbox {
-            display: none;
-            position: fixed;
-            top: 0;
-            left: 0;
-            width: 100%;
-            height: 100%;
-            background-color: rgba(0, 0, 0, 0.9);
-            z-index: 1000;
-            justify-content: center;
-            align-items: center;
-            cursor: pointer;
-        }
-        .lightbox.active {
-            display: flex;
-        }
-        .lightbox img {
-            max-width: 90%;
-            max-height: 90vh;
-            object-fit: contain;
-            border: 2px solid #ffffff;
-            box-shadow: 0 0 20px rgba(0, 0, 0, 0.5);
-        }
-        .close-button {
-            position: fixed;
-            top: 20px;
-            right: 20px;
-            color: #ffffff;
-            font-size: 30px;
-            cursor: pointer;
-            background: none;
-            border: none;
-            padding: 10px;
-            z-index: 1001;
-        }
-        @media (max-width: 768px) {
-            .gallery {
-                grid-template-columns: repeat(auto-fill, minmax(200px, 1fr));
-            }
-        }
-    </style>
-</head>
-<body>
-    <div class="gallery-container">
-    <h1>========== LS-AI-img-gen - Image Gallery ==========</h1>
-        <div class="gallery">
-            <?php
-            // Specify your images directory
-            $imageDirectory = 'images/';
-            $images = getImages($imageDirectory);
-            foreach ($images as $image) {
-                echo '<div class="gallery-item" onclick="openLightbox(\'' . $imageDirectory . $image . '\')">';
-                echo '<img src="' . $imageDirectory . $image . '" alt="' . pathinfo($image, PATHINFO_FILENAME) . '">';
-                echo '<div class="image-caption">' . pathinfo($image, PATHINFO_FILENAME) . '</div>';
-                echo '</div>';
-            }
-            if (empty($images)) {
-                echo '<p style="text-align: center; grid-column: 1/-1;">No images found in the gallery.</p>';
-            }
-            ?>
-        </div>
-    </div>
-    <!-- Lightbox container -->
-    <div class="lightbox" id="lightbox" onclick="closeLightbox()">
-        <button class="close-button" onclick="closeLightbox()">&times;</button>
-        <img id="lightbox-img" src="" alt="Lightbox image">
-    </div>
-    <script>
-        function openLightbox(imageSrc) {
-            const lightbox = document.getElementById('lightbox');
-            const lightboxImg = document.getElementById('lightbox-img');
-            lightboxImg.src = imageSrc;
-            lightbox.classList.add('active');
-            // Prevent scrolling when lightbox is open
-            document.body.style.overflow = 'hidden';
-        }
-        function closeLightbox() {
-            const lightbox = document.getElementById('lightbox');
-            lightbox.classList.remove('active');
-            // Restore scrolling
-            document.body.style.overflow = 'auto';
-        }
-        // Close lightbox with Escape key
-        document.addEventListener('keydown', function(event) {
-            if (event.key === 'Escape') {
-                closeLightbox();
-            }
-        });
-        // Prevent lightbox from closing when clicking on the image
-        document.getElementById('lightbox-img').addEventListener('click', function(event) {
-            event.stopPropagation();
-        });
-    </script>
-</body>
-</html>

 <?php
+    include 'functions.php';
+    $imageDirectory = 'images/';
+    $imageOrder = 'desc';
+    $images = getImages($imageDirectory, $imageOrder); // 'asc' or 'desc'
+    include 'gallery_content.php';
+?>

index_asc.php ADDED Viewed

	@@ -0,0 +1,8 @@

+<?php
+    include 'gallery_functions.php';
+    $imageDirectory = 'images/';
+    $imageOrder = 'asc';
+    $images = getImages($imageDirectory, $imageOrder); // 'asc' or 'desc'
+    include 'gallery_content.php';
+?>

src/img_gen copy.py DELETED Viewed

@@ -1,96 +0,0 @@
-# img_gen.py
-import sys
-import os
-import random
-from huggingface_hub import InferenceClient, login
-from datetime import datetime
-from config.config import models, prompts, api_token  # Direct import
-def generate_image(
-        prompt_alias,
-        custom_prompt,
-        characer_dropdown,
-        model_alias,
-        height=360,
-        width=640,
-        num_inference_steps=20,
-        guidance_scale=2.0,
-        seed=-1):
-    # Find the selected prompt and model
-    try:
-        prompt = next(p for p in prompts if p["alias"] == prompt_alias)["text"]
-        model_name = next(m for m in models if m["alias"] == model_alias)["name"]
-    except StopIteration:
-        return None, "ERROR: Invalid prompt or model selected."
-    # Print the original prompt and dynamic values for debugging
-    print("Original Prompt:")
-    print(prompt)
-    # Append the custom character (if provided)
-    if characer_dropdown == "Wizard":
-        prompt += " A wizard dressed in tunic figths against the beast"
-    elif characer_dropdown == "Warrior":
-        prompt += " A medieval warrior in armor figths against the beast"
-    else:
-        pass
-   # Append the custom prompt (if provided)
-    if custom_prompt and len(custom_prompt.strip()) > 0:
-        prompt += " " + custom_prompt.strip()
-    # Print the formatted prompt for debugging
-    print("\nFormatted Prompt:")
-    print(prompt)
-    # Randomize the seed if needed
-    if seed == -1:
-        seed = random.randint(0, 1000000)
-    # HF LOGIN
-    print("Initializing HF TOKEN")
-    print (api_token)
-    # login(token=api_token)
-    # print("model_name:")
-    # print(model_name)
-    # Initialize the InferenceClient
-    try:
-        print("-----INITIALIZING INFERENCE-----")
-        client = InferenceClient(model_name, token=api_token)
-        print("Inference activated")
-    except Exception as e:
-        return None, f"ERROR: Failed to initialize InferenceClient. Details: {e}"
-     #Generate the image
-    try:
-        print("-----GENERATING IMAGE-----")
-        print("-----HOLD ON-----")
-        image = client.text_to_image(
-            prompt,
-            guidance_scale=guidance_scale,
-            num_inference_steps=num_inference_steps,
-            width=width,
-            height=height,
-            seed=seed
-        )
-        print("-----IMAGE GENERATED SUCCESSFULLY!-----")
-    except Exception as e:
-        return None, f"ERROR: Failed to generate image. Details: {e}"
-    # Save the image with a timestamped filename
-    print("-----SAVING-----", image)
-    path = "images"
-    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    output_filename = f"{path}/{timestamp}_{model_alias.replace(' ', '_').lower()}_{prompt_alias.replace(' ', '_').lower()}_{characer_dropdown.replace(' ', '_').lower()}.png"
-    try:
-        image.save(output_filename)
-    except Exception as e:
-        return None, f"ERROR: Failed to save image. Details: {e}"
-    print("-----DONE!-----")
-    print("-----CALL THE BANNERS!-----")
-    return output_filename, "Image generated successfully!"

src/img_gen.py CHANGED Viewed

@@ -30,9 +30,9 @@ def generate_image(
     # Append the custom character (if provided)
     if characer_dropdown == "Wizard":
-        prompt += f" A wizard figths against the {prompt_alias}"
     elif characer_dropdown == "Warrior":
-        prompt += f" A warrior figths against the {prompt_alias}"
     else:
         pass

     # Append the custom character (if provided)
     if characer_dropdown == "Wizard":
+        prompt += f" A human wizard combat against the {prompt_alias}"
     elif characer_dropdown == "Warrior":
+        prompt += f" A human warrior figths against the {prompt_alias}"
     else:
         pass