update

app.py

@@ -43,16 +43,55 @@ by: new goals
 Distracted from: 
 """
 
+prompt5 = """Distracted from: homework
+by: side project
+Distracted from: goals
+by: new goals
+Distracted from: working hard
+by: hardly working
+Distracted from: twitter
+by: open in browser
+Distracted from:
+"""
+
+"""Distracted from: homework
+by: side project
+Distracted from: goals
+by: new goals
+Distracted from: working hard
+by: hardly working
+Distracted from: twitter
+by:  open in browser
+Distracted from: code
+by: blog post
+Distracted from: code
+by: blog post
+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))
+  
+  prompt_list = final_solution.split('\n')
+  girlfriend = prompt_list[8].split(':')[1].strip()
+  new_girl = prompt_list[9].split(':')[1].strip()
+  prompt_list.pop(0)
+  prompt_list.pop(1)
+  prompt_list = prompt_list[:7]
+  prompt_list[8] = 'Distracted from:'
+  print(f"prompt list is : {prompt_list}")
+  new_prompt = '\n'.join(prompt_list)
+  print(f"final_solution is : {new_prompt}")
+  
+  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
+  I1.text((142, 336), new_girl,font=myfont, fill =(255, 255, 255))
+  
+  return image0, new_prompt
 
 def meme_generate(img, prompt): #prompt, generated_txt): #, input_prompt_sql ): #, input_prompt_dalle2):
 
@@ -81,17 +120,17 @@ def meme_generate(img, prompt): #prompt, generated_txt): #, input_prompt_sql ):
   print(f"output_tmp is: {output_tmp}")
   solution = output_tmp.split("\nQ:")[0]   
   print(f"Final response after splits is: {solution}")
-  if '\nOutput:' in solution:
-    final_solution = solution.split("\nOutput:")[0] 
-    print(f"Response after removing output is: {final_solution}")
-  elif '\n\n' in solution:
-    final_solution = solution.split("\n\n")[0] 
-    print(f"Response after removing new line entries is: {final_solution}")
-  else:
-    final_solution = solution
+  #if '\nOutput:' in solution:
+  #  final_solution = solution.split("\nOutput:")[0] 
+  #  print(f"Response after removing output is: {final_solution}")
+  #elif '\n\n' in solution:
+  #  final_solution = solution.split("\n\n")[0] 
+  #  print(f"Response after removing new line entries is: {final_solution}")
+  #else:
+  #final_solution = solution
     
-  meme_image = write_on_image(final_solution)  
-  return meme_image, final_solution #final_solution #display_output, new_prompt  #generated_txt+prompt
+  meme_image, new_prompt = write_on_image(solution)  
+  return meme_image, solution #final_solution #display_output, new_prompt  #generated_txt+prompt
 
 
 demo = gr.Blocks()
@@ -114,8 +153,10 @@ 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():
-    in_image = gr.Image(value="./distracted0.jpg")
-    input_prompt = gr.Textbox(label="Write some prompt...", lines=5)  #input_prompt_sql
+    in_image = gr.Image(value="./distracted0.jpg", visible=False)
+    in_image_display = gr.Image(value="./distracted00.jpg", visible=True)
+    input_prompt = gr.Textbox(label="Write some prompt...", lines=5, 
+                              value = """Distracted from: homework\nby: side project\nDistracted from: goals\nby: new goals\nDistracted from: working hard\nby: hardly working\nDistracted from: twitter\nby: open in browser\nDistracted from:""", visible=False)  #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
     
@@ -125,12 +166,12 @@ with demo:
 
   with gr.Row():
     output_image = gr.Image() #type="filepath", shape=(256,256))
-    output_prompt = gr.Textbox(label="Text generated", lines=5)
+    #output_prompt = gr.Textbox(label="Text generated", lines=5)
     
   b1 = gr.Button("Generate")
   #b2 = gr.Button("Generate Image") 
 
-  b1.click(meme_generate, inputs=[in_image, input_prompt], outputs=[output_image,output_prompt]) #input_word #input_prompt_dalle2 #input_prompt_sql #example_prompt
+  b1.click(meme_generate, inputs=[in_image, input_prompt], outputs=[output_image,input_prompt])  #output_prompt
   #b2.click(poem_to_image, poem_txt, output_image)
   #examples=examples