Update app.py

app.py

@@ -4,31 +4,31 @@ from PIL import Image
 from io import BytesIO
 import os
 from huggingface_hub import InferenceClient
+
 API_TOKEN = os.getenv("HF_API_TOKEN")  # Ensure you've set this environment variable
 API_URL = "https://api-inference.huggingface.co/models/enhanceaiteam/Flux-uncensored"
 
-
-def enhanche_prompt(prompt, system_prompt='You are a prompt enhancer',  model="meta-llama/Llama-3.2-1B-Instruct", max_tokens=512, stream=False ):
-    enhancer = InferenceClient(api_key = API_TOKEN)
+def enhance_prompt(prompt, system_prompt='You are a prompt enhancer', model="meta-llama/Llama-3.2-1B-Instruct", max_tokens=512, stream=False):
+    enhancer = InferenceClient(api_key=API_TOKEN)
+    response = ""
+    
     for message in enhancer.chat_completion(
-    	model=model,
-    	messages=[
+        model=model,
+        messages=[
             {"role": "system", "content": system_prompt},
             {"role": "user", "content": prompt}
         ],
-    	max_tokens=max_tokens,
-    	stream=stream,
+        max_tokens=max_tokens,
+        stream=stream,
     ):
-        result = message.choices[0].delta.content
-        print(result, end="")
-        return result
-        
+        response += message.choices[0].delta.content
+    
+    return response.strip()  # Ensure trailing whitespace is removed
 
-# Function to call Hugging Face API and get the generated image
 def generate_image(prompt, enhance=False):
     if enhance:
-        prompt = enhanche_prompt(prompt)
-        
+        prompt = enhance_prompt(prompt)
+    
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
     data = {"inputs": prompt}
     
@@ -40,8 +40,8 @@ def generate_image(prompt, enhance=False):
         return image
     else:
         return f"Error: {response.status_code}, {response.text}"
- 
-title_html="""
+
+title_html = """
     <center>
         <div id="title-container">
             <h1 id="title-text">FLUX Capacitor</h1>
@@ -56,17 +56,17 @@ css = """
     background-repeat: no-repeat;
     background-position: center;
     background-attachment: fixed;
-    color:#000;
+    color: #000;
 }
-.dark\:bg-gray-950:is(.dark *) {
-  --tw-bg-opacity: 1;
-  background-color: rgb(157, 17, 142);
+.dark\\:bg-gray-950:is(.dark *) {
+    --tw-bg-opacity: 1;
+    background-color: rgb(157, 17, 142);
 }
 .gradio-container-4-41-0 .prose :last-child {
-  margin-top: 8px !important;
+    margin-top: 8px !important;
 }
 .gradio-container-4-41-0 .prose :last-child {
-  margin-bottom: -7px !important;
+    margin-bottom: -7px !important;
 }
 .dark {
     --button-primary-background-fill: #09e60d70;
@@ -75,29 +75,29 @@ css = """
     --background-fill-secondary: #000;
 }
 .hide-container {
-    margin-top;-2px;
+    margin-top: -2px;
 }
 #app-container3 {
-    background-color: rgba(255, 255, 255, 0.001);  /* Corrected to make semi-transparent */
+    background-color: rgba(255, 255, 255, 0.001);
     max-width: 300px;
     margin-left: auto;
     margin-right: auto;
     margin-bottom: 10px;
     border-radius: 125px;
-    box-shadow: 0 0 10px rgba(0,0,0,0.1); /* Adjusted shadow opacity */
+    box-shadow: 0 0 10px rgba(0,0,0,0.1);
 }
 #app-container {
-    background-color: rgba(255, 255, 255, 0.001);  /* Semi-transparent background */
+    background-color: rgba(255, 255, 255, 0.001);
     max-width: 300px;
-    margin: 0 auto;  /* Center horizontally */
+    margin: 0 auto;
     padding-bottom: 10px;
     border-radius: 25px;
-    box-shadow: 0 0 10px rgba(0, 0, 0, 0.1); /* Adjusted shadow opacity */
+    box-shadow: 0 0 10px rgba(0, 0, 0, 0.1);
 }
 #title-container {
     display: flex;
-    align-items: center
-    margin-bottom:10px;
+    align-items: center; /* Added semicolon */
+    margin-bottom: 10px;
     justify-content: center;
 }
 #title-icon {
@@ -112,7 +112,6 @@ css = """
 }
 """
 
-
 # Create Gradio interface
 def create_ui():
     with gr.Blocks(theme='Nymbo/Nymbo_Theme', css=css) as ui: