Update app.py

app.py

@@ -2,17 +2,26 @@ import gradio as gr
 import os
 
 stable_diffusion = gr.Blocks.load(name="spaces/runwayml/stable-diffusion-v1-5")
+clip_interrogator_2 = gr.Blocks.load(name="spaces/fffiloni/CLIP-Interrogator-2")
 
 def get_images(prompt):
     gallery_dir = stable_diffusion(prompt, fn_index=2)
-    print(gallery_dir)
-    img_results = [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
-    print(img_results)
-    print(gallery_dir)
     
+    img_results = [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
+  
     return img_results[0]
 
+def get_new_prompt(img):
+    interrogate = clip_interrogator_2(img)
+    print(interrogate)
+    return "done"
+
+def infer(prompt):
+    img = get_images(prompt)
+    result = get_new_prompt(img)
+    print(result)
+    return "prompt written"
 prompt_input = gr.Textbox(lines=4)
 gallery_output = gr.Image(type='filepath')
 
-gr.Interface(fn=get_images, inputs=[prompt_input], outputs=[gallery_output]).queue(max_size=10).launch(enable_queue=True)
+gr.Interface(fn=infer, inputs=[prompt_input], outputs=['text']).queue(max_size=10).launch(enable_queue=True)