dropdown

app.py

@@ -175,6 +175,19 @@ def reload_prompts():
     return new_dropdown
 
 # ---------------------- New Code for Loading Previous Submissions ---------------------- #
+
+def select_submission_by_id(selected_id, submissions):
+    """
+    Given a selected submission_id (as string) and the list of submissions,
+    finds the matching submission and returns its details.
+    """
+    index = next(
+        (i for i, s in enumerate(submissions) if str(s["code_review_submission_id"]) == selected_id),
+        0
+    )
+    details, code, review, index = show_submission(index, submissions)
+    return index, details, code, review
+
 def load_submissions_from_db():
     """Fetch all submissions ordered by their id in ascending order."""
     conn = get_db_connection()
@@ -212,12 +225,17 @@ def show_submission(index, submissions):
     return submission_details, input_code, ai_review_output, index
 
 
+# --- Updated load_submissions function ---
 def load_submissions():
-    """Loads submissions from the database and resets the index."""
+    """Loads submissions from the database, resets the index,
+    and returns a Dropdown update for submission choices."""
     submissions = load_submissions_from_db()
     index = 0
     details, code, review, index = show_submission(index, submissions)
-    return submissions, index, details, code, review
+    submission_choices = [str(s["code_review_submission_id"]) for s in submissions]
+    default_choice = submission_choices[0] if submission_choices else None
+    dropdown_update = gr.Dropdown(choices=submission_choices, value=default_choice)
+    return submissions, index, details, code, review, dropdown_update
 
 def next_submission(current_index, submissions):
     """Increment the index and display the next submission."""
@@ -271,30 +289,40 @@ with gr.Blocks() as demo:
         # Hidden states to store submissions and the current index
         submissions_state = gr.State([])
         submission_index_state = gr.State(0)
-        # Create two columns: left column for details and code, right column for AI review output
+        # Add a new dropdown to select submission by ID
+        submission_dropdown = gr.Dropdown(label="Select Submission by ID", choices=[], value=None)
         with gr.Row():
             back_btn = gr.Button("Back")
             next_btn = gr.Button("Next")
         with gr.Row():
             with gr.Column():
-                with gr.Accordion("Prompt Details", open= False):
+                with gr.Accordion("Prompt Details", open=False):
                     submission_details_md = gr.Markdown(label="Submission Details")
                 input_code_component = gr.Code(label="Input Code", language="python")
             with gr.Column():
                 ai_review_md = gr.Markdown(label="AI Review Output", container=True, show_copy_button=True)
         
-        # Load submissions from DB and initialize state
+        # Load submissions from DB, initialize state, and update dropdown choices
         load_submissions_btn.click(
             fn=load_submissions,
             inputs=[],
-            outputs=[submissions_state, submission_index_state, submission_details_md, input_code_component, ai_review_md]
+            outputs=[submissions_state, submission_index_state, submission_details_md, input_code_component, ai_review_md, submission_dropdown]
         )
+        
+        # When a submission is selected from the dropdown, update the submission display
+        submission_dropdown.change(
+            fn=select_submission_by_id,
+            inputs=[submission_dropdown, submissions_state],
+            outputs=[submission_index_state, submission_details_md, input_code_component, ai_review_md]
+        )
+        
         # Go to previous submission
         back_btn.click(
             fn=previous_submission,
             inputs=[submission_index_state, submissions_state],
             outputs=[submission_index_state, submission_details_md, input_code_component, ai_review_md]
         )
+        
         # Go to next submission
         next_btn.click(
             fn=next_submission,