Update app.py

app.py

@@ -10,6 +10,13 @@ from PIL import Image
 import os
 import torchvision.transforms.functional as TVF
 
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
+
+USERNAME = os.getenv("USERNAME")
+PASSWORD = os.getenv("PASSWORD")
 
 CLIP_PATH = "google/siglip-so400m-patch14-384"
 MODEL_PATH = "meta-llama/Meta-Llama-3.1-8B"
@@ -38,6 +45,11 @@ CAPTION_TYPE_MAP = {
 
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 
+def login(username, password):
+    if username == USERNAME and password == PASSWORD:
+        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(value="Login successful! You can now access the QR Code Art Generator tab.", visible=True)
+    else:
+        return gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), gr.update(value="Invalid username or password. Please try again.", visible=True)
 
 class ImageAdapter(nn.Module):
 	def __init__(self, input_features: int, output_features: int, ln1: bool, pos_emb: bool, num_image_tokens: int, deep_extract: bool):
@@ -380,14 +392,13 @@ def get_dropdown_choices(info_dict):
     return [f"{key}: {value}" for key, value in info_dict.items()]
 
 # Gradio interface
-with gr.Blocks(theme="Hev832/Applio", css=css) as demo:
+with gr.Blocks(theme="Hev832/Applio", css=css, fill_width=True, fill_height=True) as demo:
     with gr.Tab("Welcome"):
         gr.Markdown(
             """
-            <img src="https://cdn-uploads.huggingface.co/production/uploads/64740cf7485a7c8e1bd51ac9/8hAzfeJH9xMj8Sa9BIy-f.png" alt="UDG" class="centered-image" width="250" style="max-width: 100%; height: auto;">
-
+            <img src="https://cdn-uploads.huggingface.co/production/uploads/64740cf7485a7c8e1bd51ac9/LVZnwLV43UUvKu3HORqSs.webp" alt="UDG" class="centered-image" width="250" style="max-width: 100%; height: auto;">
 
-            # 🎨 Underground Dgitial's Caption Captain: AI-Powered Art Inspiration
+            # 🎨 Underground Digital's Caption Captain: AI-Powered Art Inspiration
 
             ## Accelerate Your Creative Workflow with Intelligent Image Analysis
 
@@ -401,8 +412,16 @@ with gr.Blocks(theme="Hev832/Applio", css=css) as demo:
             4. **Generate and Iterate**: Click 'Caption' to analyze your image and use the results to inspire new creations.
             """
         )
+
+        with gr.Row():
+            username = gr.Textbox(label="Username", placeholder="Enter your username", value="ugd")
+        with gr.Row():
+            password = gr.Textbox(label="Password", type="password", placeholder="Enter your password", value="ugd!")
+        with gr.Row():
+            login_button = gr.Button("Login", size="sm")
+        login_message = gr.Markdown(visible=False)
     
-    with gr.Tab("Caption Captain"):
+    with gr.Tab("Caption Captain") as app_container:
         with gr.Accordion("How to Use Caption Captain", open=False):
             gr.Markdown("""
             # How to Use Caption Captain