Update app.py

app.py

@@ -5,14 +5,9 @@ from PIL import Image
 import requests
 import io
 import time
-import whisper
-import gradio as gr
 from groq import Groq
-from deep_translator import GoogleTranslator
 import torch
 
-
-# Replace with your actual Hugging Face API details
 os.environ['hugging']
 H_key = os.getenv('hugging')
 API_URL = "https://api-inference.huggingface.co/models/Artples/LAI-ImageGeneration-vSDXL-2"
@@ -94,17 +89,56 @@ def process_audio_or_text(input_text, audio_path, generate_image_flag):
 
     if generate_image_flag:  # Generate image if the checkbox is checked
         image = generate_image(translation)
-    
+
     return tamil_text, chatbot_response, image  # Return both chatbot response and image (if generated)
 
-with gr.Blocks() as iface:
-    gr.Markdown("# AI Chatbot and Image Generation App")
+# Custom CSS for improved styling and centered title
+css = """
+    .gradio-container {
+        font-family: 'Georgia', serif;
+        background-color: #f5f5f5;
+        padding: 20px;
+        color: #000000;
+    }
+    .gr-row {
+        box-shadow: 0px 4px 12px rgba(0, 0, 0, 0.1);
+        background-color: #ffffff;
+        border-radius: 10px;
+        padding: 20px;
+        margin: 10px 0;
+    }
+    .gr-button {
+        background-color: #8b4513;
+        color: white;
+        font-size: 16px;
+        border-radius: 5px;
+    }
+    .gr-button:hover {
+        background-color: #6a3511;
+    }
+    .gr-checkbox-label {
+        font-weight: bold;
+    }
+    .gr-image {
+        border-radius: 10px;
+        box-shadow: 0px 4px 12px rgba(0, 0, 0, 0.1);
+    }
+    #main-title {
+        text-align: center;
+        font-size: 28px;
+        font-weight: bold;
+        color: #8b4513;
+    }
+"""
+
+with gr.Blocks(css=css) as iface:
+    gr.Markdown("<h1 id='main-title'>🖼️ AI Chatbot and Image Generation App</h1>")
 
     with gr.Row():
         with gr.Column(scale=1):  # Left side (Inputs and Buttons)
             user_input = gr.Textbox(label="Enter Tamil text", placeholder="Type your message here...")
-            audio_input = gr.Audio(type="filepath", label=" Or upload audio (for Image Generation)")
-            image_generation_checkbox = gr.Checkbox(label="Generate Image", value=False)
+            audio_input = gr.Audio(type="filepath", label="Or upload audio (for Image Generation)")
+            image_generation_checkbox = gr.Checkbox(label="Generate Image", value=True)
 
             # Buttons
             submit_btn = gr.Button("Submit")
@@ -119,7 +153,7 @@ with gr.Blocks() as iface:
     submit_btn.click(fn=process_audio_or_text,
                      inputs=[user_input, audio_input, image_generation_checkbox],
                      outputs=[text_output_1, text_output_2, image_output])
-    
+
     clear_btn.click(lambda: ("", None, False, "", "", None),
                     inputs=[],
                     outputs=[user_input, audio_input, image_generation_checkbox, text_output_1, text_output_2, image_output])