Update app.py

app.py

@@ -1,4 +1,3 @@
-import numpy as np
 import gradio as gr
 import requests
 import time
@@ -6,12 +5,11 @@ import json
 import base64
 import os
 from io import BytesIO
-import PIL
-from PIL.ExifTags import TAGS
 import html
 import re
 
 
+
 class Prodia:
     def __init__(self, api_key, base=None):
         self.base = base or "https://api.prodia.com/v1"
@@ -83,6 +81,7 @@ def image_to_base64(image):
 
     return img_str.decode('utf-8')  # Convert bytes to string
 
+
 def remove_id_and_ext(text):
     text = re.sub(r'\[.*\]$', '', text)
     extension = text[-12:].strip()
@@ -92,6 +91,7 @@ def remove_id_and_ext(text):
         text = text[:-4]
     return text
 
+
 def get_data(text):
     results = {}
     patterns = {
@@ -119,9 +119,10 @@ def get_data(text):
         results['h'] = None
     return results
 
+
 def send_to_txt2img(image):
-    
-    result = {tabs: gr.Tabs.update(selected="t2i")}
+
+    result = {tabs: gr.update(selected="t2i")}
 
     try:
         text = image.info['parameters']
@@ -142,15 +143,6 @@ def send_to_txt2img(image):
 
     except Exception as e:
         print(e)
-        result[prompt] = gr.update()
-        result[negative_prompt] = gr.update()
-        result[steps] = gr.update()
-        result[seed] = gr.update()
-        result[cfg_scale] = gr.update()
-        result[width] = gr.update()
-        result[height] = gr.update()
-        result[sampler] = gr.update()
-        result[model] = gr.update()
 
         return result
 
@@ -163,6 +155,7 @@ for model_name in model_list:
     name_without_ext = remove_id_and_ext(model_name)
     model_names[name_without_ext] = model_name
 
+
 def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
     result = prodia_client.generate({
         "prompt": prompt,
@@ -180,6 +173,7 @@ def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, he
 
     return job["imageUrl"]
 
+
 def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
     result = prodia_client.transform({
         "imageData": image_to_base64(input_image),