update api

app.py

@@ -6,14 +6,21 @@ import time
 import os
 from PIL import Image
 
-# Get SR API endpoint from environment
-ENHANCE_API = os.getenv("img_enhance_api")
+# Get API endpoints from environment
+SR_API = os.getenv("sr_api")
+ENHANCE_API = os.getenv("enhance_api")
 
-def apply_super_resolution(input_image, scale_factor):
+def apply_super_resolution(input_image, scale_factor, api_choice):
     """Apply super-resolution to input image"""
     if input_image is None:
         return None, "❌ Please upload an image first", gr.update(visible=False)
     
+    # Select API endpoint based on user choice
+    api_endpoint = SR_API if api_choice == "SR API" else ENHANCE_API
+    
+    if not api_endpoint:
+        return None, f"❌ {api_choice} endpoint not configured", gr.update(visible=False)
+    
     try:
         # Convert PIL image to bytes
         img_buffer = io.BytesIO()
@@ -22,7 +29,7 @@ def apply_super_resolution(input_image, scale_factor):
         
         # Call super-resolution API
         response = requests.post(
-            f"{ENHANCE_API}/invocations",
+            f"{api_endpoint}/invocations",
             headers={"Content-Type": "application/octet-stream"},
             data=img_bytes,
             params={
@@ -44,8 +51,9 @@ def apply_super_resolution(input_image, scale_factor):
             # Create download file
             timestamp = int(time.time())
             download_path = f"enhanced_image_{timestamp}.png"
+            enhanced_image.save(download_path, format='PNG')
             
-            status = f"✅ Enhancement successful!\n"
+            status = f"✅ Enhancement successful using {api_choice}!\n"
             status += f"Model: {result.get('model', 'RealESRGAN_x4plus')}\n"
             status += f"Scale: {result.get('outscale', 4.0)}x\n"
             status += f"Input: {result.get('input_img_width', 0)}x{result.get('input_img_height', 0)}\n"
@@ -75,6 +83,14 @@ def main():
             
             with gr.Column(scale=1):
                 gr.Markdown("### Enhancement Settings")
+                
+                api_dropdown = gr.Dropdown(
+                    choices=["SR API", "ENHANCE API"], 
+                    value="SR API", 
+                    label="API Choice",
+                    info="Choose which enhancement API to use"
+                )
+                
                 scale_dropdown = gr.Dropdown(
                     choices=[1, 2, 4], 
                     value=4, 
@@ -117,7 +133,7 @@ def main():
         # Event handlers
         enhance_button.click(
             fn=apply_super_resolution,
-            inputs=[input_image, scale_dropdown],
+            inputs=[input_image, scale_dropdown, api_dropdown],
             outputs=[image_slider, status_text, download_button],
             show_progress=True
         )