a

app.py

@@ -1,6 +1,10 @@
 import gradio as gr
 import requests
 import os 
+import PIL
+from PIL import Image
+from PIL import ImageDraw
+from PIL import ImageFont
 
 ##Bloom
 API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
@@ -30,11 +34,26 @@ prompt3 = """Q: A juggler can juggle 16 balls. Half of the balls are golf balls,
 A: Let’s think step by step.
 """
 
+prompt = """Distracted from hubble by james webb.
+Distracted from homework by side projeect.
+Distracted from goals by new goals.
+Distracted from 
+"""
 
+def write_on_image(final_solution):
+  print("************ Inside write_on_image ***********")
+  image_path0 = "./distracted0.jpg"
+  image0 = Image.open(image_path0)
+  I1 = ImageDraw.Draw(image0)
+  myfont = ImageFont.truetype('./font1.ttf', 30)
+  I1.text((613, 89), "girlfriend",font=myfont, fill =(255, 255, 255))
+  I1.text((371, 223), "ME", font=myfont, fill =(255, 255, 255))
+  I1.text((142, 336), "new girl",font=myfont, fill =(255, 255, 255))
+  return image0
 
-def text_generate(prompt, generated_txt): #, input_prompt_sql ): #, input_prompt_dalle2):
+def meme_generate(img): #prompt, generated_txt): #, input_prompt_sql ): #, input_prompt_dalle2):
 
-  print(f"*****Inside text_generate - Prompt is :{prompt}")
+  print(f"*****Inside meme_generate - Prompt is :{prompt}")
   #if input_prompt_sql != '':
   #  prompt = input_prompt_sql #"Instruction: Given an input question, respond with syntactically correct PostgreSQL\nInput: " +input_prompt_sql + "\nPostgreSQL query: "
   #elif input_prompt_dalle2 !='':
@@ -93,8 +112,9 @@ def text_generate(prompt, generated_txt): #, input_prompt_sql ): #, input_prompt
       new_prompt = display_output[-first_period_loc:-1]
       print(f"Not sending blank as prompt so new prompt for next cycle is : {new_prompt}")
     display_output = display_output[:-1]
-    
-  return display_output, new_prompt  #generated_txt+prompt #final_solution
+  
+  meme_image = write_on_image(final_solution)  
+  return meme_image #final_solution #display_output, new_prompt  #generated_txt+prompt
 
 
 demo = gr.Blocks()
@@ -102,7 +122,7 @@ demo = gr.Blocks()
 with demo:
   gr.Markdown("<h1><center>Bloom</center></h1>")
   gr.Markdown(
-        """Testing Bloom for SQL generation """
+        """Testing Bloom. WIP"""
         )
   with gr.Row():
     #example_prompt = gr.Radio( ["Q: A juggler can juggle 16 balls. Half of the balls are golf balls, and half of the golf balls are blue. How many blue golf balls are there?\nA: Let’s think step by step.\n"], label= "Choose a sample Prompt")
@@ -117,19 +137,20 @@ with demo:
     #"Dalle Prompt: Cyberwave vaporpunk art of a kneeling figure, looking up at a glowing neon book icon, smoke and mist, pink and blue lighting, cybernetic sci-fi render\nNew Dalle Prompt: " ], label= "Choose a sample Prompt")
     
   #with gr.Row():
-    input_prompt = gr.Textbox(label="Write some text to get started...", lines=3)  #input_prompt_sql
+    in_image = gr.Image(value="./distracted0.jpg")
+    #input_prompt = gr.Textbox(label="Write some text to get started...", lines=3)  #input_prompt_sql
     #input_prompt_dalle2 = gr.Textbox(label="Or Write sample Dalle2 prompts to get more Prompt ideas...")
     #input_prompt2 = gr.Textbox(label="Write some text to get started...", lines=3, visible=False)  #input_prompt_sql
     
     #input_word = gr.Textbox(placeholder="Enter a word here to generate text ...")
-  with gr.Row():
-    generated_txt = gr.Textbox(lines=7, visible = True)
-    #output_image = gr.Image(type="filepath", shape=(256,256))
+  #with gr.Row():
+    #generated_txt = gr.Textbox(lines=7, visible = True)
+    output_image = gr.Image() #type="filepath", shape=(256,256))
   
-  b1 = gr.Button("Generate Text")
+  b1 = gr.Button("Generate MeMe")
   #b2 = gr.Button("Generate Image") 
 
-  b1.click(text_generate, inputs=[input_prompt, generated_txt], outputs=[generated_txt, input_prompt]) #input_word #input_prompt_dalle2 #input_prompt_sql #example_prompt
+  b1.click(text_generate, inputs=in_image, outputs=output_image) #input_word #input_prompt_dalle2 #input_prompt_sql #example_prompt
   #b2.click(poem_to_image, poem_txt, output_image)
   #examples=examples