eli02 commited on
Commit
767e056
·
1 Parent(s): 28cc574

update: Modify reaction collection logic in Gradio interface to only save provided reactions

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -136,10 +136,13 @@ def collect_reactions(results, reaction_1, reaction_2, reaction_3):
136
  # Define the Gradio interface
137
  def gradio_interface(username, password, user_type, query, reaction_1=None, reaction_2=None, reaction_3=None):
138
  time_taken, results, save_message = main(username, password, user_type, query)
139
- if results:
 
 
140
  reactions = collect_reactions(results, reaction_1, reaction_2, reaction_3)
141
  if any(reactions.values()): # If any reaction is provided, save them
142
  _, _, save_message = main(username, password, user_type, query, reactions)
 
143
  return time_taken, results, save_message
144
 
145
  # Input and output components for Gradio
 
136
  # Define the Gradio interface
137
  def gradio_interface(username, password, user_type, query, reaction_1=None, reaction_2=None, reaction_3=None):
138
  time_taken, results, save_message = main(username, password, user_type, query)
139
+
140
+ # Only collect reactions if they are provided
141
+ if reaction_1 is not None or reaction_2 is not None or reaction_3 is not None:
142
  reactions = collect_reactions(results, reaction_1, reaction_2, reaction_3)
143
  if any(reactions.values()): # If any reaction is provided, save them
144
  _, _, save_message = main(username, password, user_type, query, reactions)
145
+
146
  return time_taken, results, save_message
147
 
148
  # Input and output components for Gradio