Update app.py

app.py

@@ -46,7 +46,7 @@ def initialize_hf_dataset():
             with open(TEMP_JSON, "w") as f:
                 pass
         
-        return True, f"Connected to HuggingFace dataset: {repo_id}"
+        return True, f"{repo_id}"
     except Exception as e:
         return False, f"Error initializing HuggingFace dataset: {str(e)}"
 
@@ -77,7 +77,7 @@ def push_to_hf_dataset(data_row):
     except Exception as e:
         return False, f"Error pushing to HuggingFace: {str(e)}"
 
-def save_choice(text_id, original_text, summary_a, summary_b, choice, notes=""):
+def save_choice(text_id, original_text, summary_a, summary_b, choice, notes="", request_id=""):
     """Save the user's choice locally and to HuggingFace dataset"""
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     chosen_summary = "A" if choice == "Summary A" else "B"
@@ -90,7 +90,8 @@ def save_choice(text_id, original_text, summary_a, summary_b, choice, notes=""):
         "summary_a": summary_a,
         "summary_b": summary_b,
         "chosen_summary": chosen_summary,
-        "notes": notes
+        "notes": notes,
+        "request_id": request_id
     }
     
     # Save locally
@@ -105,10 +106,12 @@ def save_choice(text_id, original_text, summary_a, summary_b, choice, notes=""):
     # Push to HuggingFace
     success, message = push_to_hf_dataset(new_row)
     
+    request_id_msg = f" (Request ID: {request_id})" if request_id else ""
+    
     if success:
-        return f"Selection saved for text ID: {text_id}! You chose {'Summary A' if choice == 'Summary A' else 'Summary B'}. Pushed to HuggingFace."
+        return f"Selection saved for text ID: {text_id}{request_id_msg}! You chose {'Summary A' if choice == 'Summary A' else 'Summary B'}. Pushed to HuggingFace."
     else:
-        return f"Selection saved locally for text ID: {text_id}. HuggingFace push failed: {message}"
+        return f"Selection saved locally for text ID: {text_id}{request_id_msg}. HuggingFace push failed: {message}"
 
 class SummaryChooser:
     def __init__(self):
@@ -116,6 +119,17 @@ class SummaryChooser:
         self.current_index = 0
         self.total_items = len(self.df)
         self.hf_status = initialize_hf_dataset()
+        self.request_id = ""  # Initialize empty request ID
+    
+    def set_request_id(self, request: gr.Request):
+        """Set the request ID from the URL query parameters"""
+        try:
+            query_params = request.query_params
+            self.request_id = query_params.get("id", "")
+            return f"Request ID: {self.request_id}" if self.request_id else "No Request ID provided"
+        except:
+            self.request_id = ""
+            return "Failed to get Request ID"
     
     def get_current_item(self):
         """Get the current item from the dataframe"""
@@ -134,8 +148,8 @@ class SummaryChooser:
         # Get current values
         text_id, text, summary_a, summary_b, _ = self.get_current_item()
         
-        # Save the choice
-        result_message = save_choice(text_id, text, summary_a, summary_b, choice, notes)
+        # Save the choice with the request ID
+        result_message = save_choice(text_id, text, summary_a, summary_b, choice, notes, self.request_id)
         
         # Move to next item or wrap around
         self.current_index = (self.current_index + 1) % self.total_items
@@ -159,7 +173,7 @@ class SummaryChooser:
     def get_hf_status(self):
         """Get the status of HuggingFace integration"""
         success, message = self.hf_status
-        return f"HuggingFace Status: {'Connected' if success else 'Not Connected'} - {message}"
+        return f"{'Connected' if success else 'Not Connected'} - {message}"
 
 # Create the application
 app = SummaryChooser()
@@ -171,9 +185,14 @@ with gr.Blocks(title="Summary Chooser") as interface:
     
     with gr.Row():
         with gr.Column():
-            progress_label = gr.Label(label="Progress")  
+            progress_label = gr.Label(label="Progress")
+    
         with gr.Column():
             hf_status = gr.Label(label="HuggingFace Status", value=app.get_hf_status())
+        
+        with gr.Column():
+            request_id_label = gr.Label(label="Request ID")
+        
         with gr.Column():
             text_id_box = gr.Textbox(label="Text ID", interactive=False)
     
@@ -222,6 +241,13 @@ with gr.Blocks(title="Summary Chooser") as interface:
         inputs=[],
         outputs=[text_id_box, text_box, summary_a, summary_b, progress_label, result_box]
     )
+    
+    # Load the request ID from the URL when the page loads
+    interface.load(
+        fn=app.set_request_id,
+        inputs=[],
+        outputs=[request_id_label]
+    )
 
 # Launch the application
 if __name__ == "__main__":