Update app.py

app.py

@@ -9,6 +9,11 @@ def display_df():
   df_images = df.head(1000)
   return df_images
 
+def display_df_search(search):
+  df_images = df.loc[df['prompt'].str.contains(search, case=False, na=False)]
+  #df_images = df.head(1000)
+  return df_images
+
 def display_next1000(dataframe, end):
   dataframe = dataframe.sample(frac=1)
   start = (end  or dataframe.index[-1]) + 1
@@ -16,20 +21,33 @@ def display_next1000(dataframe, end):
   df_images = df.loc[start:end]
   return df_images, end
 
+def Match(name):
+    pd.set_option("display.max_rows", None)
+    data = df
+    swith=data.loc[data['prompt'].str.contains(name, case=False, na=False)]
+    return swith
+  
     
 with gr.Blocks() as demo:
   gr.Markdown("<h1><center>🍰PrompTart🎨</center></h1>")
   gr.Markdown("""<div align="center">Art Prompts from <a href = "https://playgroundai.com/">Playground</a>. <a href="https://github.com/playgroundai/liked_images">Git</a>.  <a href="https://playgroundai.com/create">Create Art Here</a>.  <a href="https://paperswithcode.com/datasets?q=art&v=lst&o=newest">Papers,Code,Datasets for SOTA in Art</a>""")
+  
+
 
   with gr.Row():
     num_end = gr.Number(visible=False)
     b1 = gr.Button("Images and Prompts 0-1000")
     b2 = gr.Button("Next 1000 Images and Prompts")
     
+  with gr.Row(): # inputs and buttons
+    inp = gr.Textbox(lines=1, default="", label="Search")
+    b3 = gr.Button("Search")    
+    
   with gr.Row():
     out_dataframe = gr.Dataframe(wrap=True, max_rows=1000, overflow_row_behaviour= "paginate", datatype = ["markdown", "markdown"], headers=['url', 'prompt'])
     
   b1.click(fn=display_df, outputs=out_dataframe) 
   b2.click(fn=display_next1000, inputs= [out_dataframe, num_end ], outputs=[out_dataframe, num_end])
+  b3.click(fn=display_df_search, inputs=inp, outputs=out_dataframe) 
 
 demo.launch(debug=True, show_error=True)