Remove unused environment variable and enhance random number generator initialization; delete obsolete prompt enhancement function and related UI elements

app.py

@@ -29,10 +29,8 @@ import gradio as gr
 from models.infinity import Infinity
 from models.basic import *
 from utils.dynamic_resolution import dynamic_resolution_h_w, h_div_w_templates
-from gradio_client import Client
 
 torch._dynamo.config.cache_size_limit = 64
-client = Client("Qwen/Qwen2.5-72B-Instruct")
 
 # Define a function to download weights if not present
 def download_infinity_weights(weights_path):
@@ -378,60 +376,6 @@ def load_transformer(vae, args):
     )
     return infinity
 
-def enhance_prompt(prompt):
-    SYSTEM = """You are part of a team of bots that creates images. You work with an assistant bot that will draw anything you say.  
-
-When given a user prompt, your role is to transform it into a creative, detailed, and vivid image description that focuses on visual and sensory features. Avoid directly referencing specific real-world people, places, or cultural knowledge unless explicitly requested by the user.  
-
-### Guidelines for Generating the Output:  
-
-1. **Output Format:**  
-   Your response must be in the following dictionary format:  
-   ```json
-   {
-     "prompt": "<enhanced image description>",
-     "cfg": <cfg value>
-   }
-   ```  
-
-2. **Enhancing the "prompt" field:**  
-   - Use your creativity to expand short or vague prompts into highly detailed, visually rich descriptions.  
-   - Focus on describing visual and sensory elements, such as colors, textures, shapes, lighting, and emotions.  
-   - Avoid including known real-world information unless the user explicitly requests it. Instead, describe features that evoke the essence or appearance of the scene or subject.  
-   - For particularly long user prompts (over 50 words), output them directly without refinement.  
-   - Image descriptions must remain between 8-512 words. Any excess text will be ignored.  
-   - If the user's request involves rendering specific text in the image, enclose that text in single quotation marks and prefix it with "the text".  
-
-3. **Determining the "cfg" field:**  
-   - If the image to be generated is likely to feature a clear face, set `"cfg": 1`.  
-   - If the image does not prominently feature a face, set `"cfg": 3`.  
-
-4. **Examples of Enhanced Prompts:**  
-   - **User prompt:** "a tree"  
-     **Enhanced prompt:** "A towering tree with a textured bark of intricate ridges and grooves stands under a pale blue sky. Its sprawling branches create an umbrella of rich, deep green foliage, with a few golden leaves scattered, catching the sunlight like tiny stars."  
-     **Cfg:** `3`  
-
-   - **User prompt:** "a person reading"  
-     **Enhanced prompt:** "A figure sits on a cozy armchair, illuminated by the soft, warm glow of a nearby lamp. Their posture is relaxed, and their hands gently hold an open book. Shadows dance across their thoughtful expression, while the fabric of their clothing appears textured and soft, with subtle folds."  
-     **Cfg:** `1`  
-
-5. **Your Output:**  
-   Always return a single dictionary containing both `"prompt"` and `"cfg"` fields. Avoid any additional commentary or explanations.  
-
-Don't write anything except the dictionary in the output. (Don't start with ```)
-"""
-    result = client.predict(
-            query=prompt,
-            history=[],
-            system=SYSTEM,
-            api_name="/model_chat"
-    )
-
-    dict_of_inputs = json.loads(result[1][-1][-1])
-    print(dict_of_inputs)
-
-    return gr.update(value=dict_of_inputs["prompt"]), gr.update(value=float(dict_of_inputs['cfg']))
-
 # Set up paths
 weights_path = Path(__file__).parent / 'weights'
 weights_path.mkdir(exist_ok=True)
@@ -554,7 +498,6 @@ with gr.Blocks() as demo:
             # Prompt Settings
             gr.Markdown("### Prompt Settings")
             prompt = gr.Textbox(label="Prompt", value="alien spaceship enterprise", placeholder="Enter your prompt here...")
-            enhance_prompt_button = gr.Button("Enhance Prompt", variant="secondary")
 
             # Image Settings
             gr.Markdown("### Image Settings")
@@ -575,13 +518,6 @@ with gr.Blocks() as demo:
     
     # Error Handling
     error_message = gr.Textbox(label="Error Message", visible=False)
-    
-    # Link the enhance prompt button to the prompt enhancement function
-    enhance_prompt_button.click(
-        enhance_prompt,
-        inputs=prompt,
-        outputs=[prompt, cfg],
-    )
 
     # Link the generate button to the image generation function
     generate_button.click(