Added AI gen code

app.py

@@ -4,8 +4,14 @@ import warnings
 import random
 import time
 import logging
+import dotenv
+import fal_client
+import requests
 from typing import Dict, List, Tuple, Union, Optional
 
+dotenv.load_dotenv()
+FAL_KEY = os.getenv("FAL_KEY")
+
 # Configure logging
 logging.basicConfig(
     level=logging.INFO,
@@ -218,12 +224,6 @@ def draw_box(box: torch.Tensor, draw: ImageDraw.Draw, label: Optional[str]) -> N
 
 def run_grounded_sam(
     input_image
-    # text_prompt: str, 
-    # task_type: str, 
-    # box_threshold: float, 
-    # text_threshold: float, 
-    # iou_threshold: float, 
-    # hq_token_only
 ) -> List[Image.Image]:
     """Main function to run GroundingDINO and SAM-HQ"""
     try:
@@ -328,6 +328,31 @@ def run_grounded_sam(
         else:
             return [Image.new('RGB', (400, 300), color='gray'), Image.new('RGBA', (400, 300), color=(0, 0, 0, 0))]
 
+def generate_ai_bg(input_img, prompt):
+    print(FAL_KEY)
+    handler = fal_client.submit(
+        "fal-ai/iclight-v2",
+        arguments={
+            "prompt": prompt,
+            "image_url": input_img
+        },
+        webhook_url="https://optional.webhook.url/for/results",
+    )
+
+    request_id = handler.request_id
+
+    status = fal_client.status("fal-ai/iclight-v2", request_id, with_logs=True)
+
+    result = fal_client.result("fal-ai/iclight-v2", request_id)
+
+    ic_light_img = result['images'][0]['url']
+
+    return ic_light_img
+
+def generate_image(input_img, prompt):
+    ai_gen_image = generate_ai_bg(input_img, prompt)
+
+    return ai_gen_image
 
 def create_ui():
     """Create Gradio UI for CarViz demo"""
@@ -339,7 +364,7 @@ def create_ui():
         with gr.Row():
             with gr.Column():
                 input_image = gr.Image(type="pil", label="image")
-
+                prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
                 run_button = gr.Button(value='Run')
 
             with gr.Column():
@@ -351,7 +376,8 @@ def create_ui():
         run_button.click(
             fn=run_grounded_sam, 
             inputs=[
-                input_image
+                input_image,
+                prompt
             ], 
             outputs=gallery
         )     
@@ -360,7 +386,7 @@ def create_ui():
 
 
 if __name__ == "__main__":
-    parser = argparse.ArgumentParser("Grounded SAM demo", add_help=True)
+    parser = argparse.ArgumentParser("Carviz demo", add_help=True)
     parser.add_argument("--debug", action="store_true", help="using debug mode")
     parser.add_argument("--share", action="store_true", help="share the app")
     parser.add_argument('--no-gradio-queue', action="store_true", help="disable gradio queue")

requirements.txt

@@ -19,3 +19,5 @@ torchvision
 transformers
 yapf
 scipy
+dotenv
+fal_client