rthrth

app.py

@@ -63,25 +63,17 @@ def perform_query(query, authors, num_results):
     results = query_chroma(embeddings, authors, num_results)
 
     if "error" in results:
-        return [gr.update(visible=True, value=f"Error: {results['error']}") for _ in range(max_textboxes * 3)]
+        return [gr.update(visible=True, value=f"Error: {results['error']}") for _ in range(max_textboxes * 2)]
 
     updates = []
     for res in results:
-        markdown_content = f"**{res['author']}, {res['book']}, Distance: {res['distance']}**\n\n{res['text']}"
+        markdown_content = f"**{res['author']}, {res['book']}**\n\n{res['text']}"
         updates.append(gr.update(visible=True, value=markdown_content))
-        updates.append(gr.update(visible=True, value="Flag", elem_id=f"flag-{len(updates)//2}"))
-        updates.append(gr.update(visible=False, value=res['id']))  # Hide the ID textbox
-    
-    updates += [gr.update(visible=False)] * (3 * (max_textboxes - len(results)))
 
-    return updates
+    updates += [gr.update(visible=False)] * (max_textboxes - len(results))
 
-# Initialize the CSVLogger callback for flagging 
-callback = gr.CSVLogger()
+    return updates
 
-def flag_output(query, output_text, output_id):
-    callback.flag([query, output_text, output_id])
-        
 # Gradio interface
 max_textboxes = 30
 
@@ -96,34 +88,18 @@ with gr.Blocks(css=".custom-markdown { border: 1px solid #ccc; padding: 10px; bo
 
     components = []
     textboxes = []
-    flag_buttons = []
-    ids = []
 
     for _ in range(max_textboxes):
         with gr.Column() as col:
             text_out = gr.Markdown(visible=False, elem_classes="custom-markdown")
-            flag_btn = gr.Button(value="Flag", visible=False)
-            id_out = gr.Textbox(visible=False)
-            components.extend([text_out, flag_btn, id_out])
+            components.append(text_out)
             textboxes.append(text_out)
-            flag_buttons.append(flag_btn)
-            ids.append(id_out)
 
-    callback.setup([inp] + textboxes + ids, "flagged_data_points")
-        
     btn.click(
         fn=perform_query, 
         inputs=[inp, author_inp, num_results_inp],
         outputs=components
     )
     
-    for i in range(0, len(components), 3):
-        flag_buttons[i//3].click(
-            fn=flag_output, 
-            inputs=[inp, textboxes[i//3], ids[i//3]], 
-            outputs=[],
-            preprocess=False
-        )
-
 
 demo.launch()