Update app.py

app.py

@@ -18,27 +18,21 @@ from deep_translator import GoogleTranslator
 from datetime import datetime
 from theme import theme
 from typing import Tuple
-from mistralai import Mistral
 from fastapi import FastAPI
 
 app = FastAPI()
 
 
+
 API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 100
 
-api_key = os.getenv("MISTRAL_KEY")
-Mistralclient = Mistral(api_key=api_key)
-
 def flip_image(x):
     return np.fliplr(x)
 
 def clear():
-  return None 
-
-def change_tab():
-    return gr.Tabs.update(selected=1)    
+  return None    
 
 def query(lora_id, prompt, is_negative=False, steps=28, cfg_scale=3.5, sampler="DPM++ 2M Karras", seed=-1, strength=100, width=896, height=1152):
     if prompt == "" or prompt == None:
@@ -103,85 +97,18 @@ examples = [
     "a beautiful woman with black hair and brown eyes",
 ]
 
-def encode_image(image_path):
-    """Encode the image to base64."""
-    try:
-        # Open the image file
-        image = Image.open(image_path).convert("RGB")
-
-        # Resize the image to a height of 512 while maintaining the aspect ratio
-        base_height = 512
-        h_percent = (base_height / float(image.size[1]))
-        w_size = int((float(image.size[0]) * float(h_percent)))
-        image = image.resize((w_size, base_height), Image.LANCZOS)
-
-        # Convert the image to a byte stream
-        buffered = BytesIO()
-        image.save(buffered, format="JPEG")
-        img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
-
-        return img_str
-    except FileNotFoundError:
-        print(f"Error: The file {image_path} was not found.")
-        return None
-    except Exception as e:  # Add generic exception handling
-        print(f"Error: {e}")
-        return None
-
-def feifeichat(image):
-    try:
-        model = "pixtral-large-2411"
-        # Define the messages for the chat
-        base64_image = encode_image(image)
-        messages = [{
-            "role":
-            "user",
-            "content": [
-                {
-                    "type": "text",
-                    "text": "Please provide a detailed description of this photo"
-                },
-                {
-                    "type": "image_url",
-                    "image_url": f"data:image/jpeg;base64,{base64_image}" 
-                },
-            ],
-            "stream": False,
-        }]
-    
-        partial_message = ""
-        for chunk in Mistralclient.chat.stream(model=model, messages=messages):
-            if chunk.data.choices[0].delta.content is not None:
-                partial_message = partial_message + chunk.data.choices[
-                    0].delta.content
-                yield partial_message
-    except Exception as e:  # 添加通用异常处理
-        print(f"Error: {e}")
-        return "Please upload a photo"
-
-
 css = """
-footer{display:none !important}
-
-
+#app-container {
+    max-width: 930px;
+    margin-left: auto;
+    margin-right: auto;
+}
+".gradio-container {background: url('file=abstract.jpg')}
    
 """
 with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
     gr.HTML("<center><h6>🎨 FLUX.1-Dev with LoRA 🇬🇧</h6></center>")
-    with gr.Tabs() as tabs:
-        with gr.TabItem(label="Image To Prompt", visible=True, id=1):
-            with gr.Row():
-                with gr.Column():
-                    input_img = gr.Image(label="Input Picture 🖼️",height=320,type="filepath")
-                    submit_btn = gr.Button(value="Submit", variant='primary')
-                with gr.Column():
-                    output_text = gr.Textbox(label="Flux Prompt ✍️", show_copy_button = True)
-                    clr_button =gr.Button("Clear 🗑️ ",variant="primary", elem_id="clear_button")
-                    clr_button.click(lambda: (None, None), None, [input_img, output_text], queue=False, show_api=False)
-        
-        submit_btn.click(feifeichat, [input_img], [output_text])
-    
-    with gr.TabItem(label="Text to Image", visible=True, id=0):
+    with gr.Tab("Text to Image"):
         with gr.Column(elem_id="app-container"):
             with gr.Row():
                 with gr.Column(elem_id="prompt-container"):
@@ -254,7 +181,7 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
                     
                 text_button.click(query, inputs=[custom_lora, text_prompt, negative_prompt, steps, cfg, method, seed, strength, width, height], outputs=[image_output, seed_output])
 
-    with gr.TabItem(label="Flip Image", visible=True, id=2):
+    with gr.Tab("Flip Image"):
         with gr.Row():
             image_input = gr.Image()
             image_output = gr.Image(format="png")
@@ -262,7 +189,7 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
             image_button = gr.Button("Run", variant='primary')
             image_button.click(flip_image, inputs=image_input, outputs=image_output, concurrency_limit=2)
 
-    with gr.TabItem(label="Tips", visible=True, id=3):
+    with gr.Tab("Tips"):
         with gr.Row():
                 gr.Markdown(
         """