Update app.py

app.py

@@ -75,7 +75,7 @@ def update_state_dropdown(country):
         return gr.Dropdown(choices=states_by_country[country], label=f"State/Province in {country}:", interactive=True)
     return gr.Dropdown(choices=[], label="State/Province:", interactive=True)
 
-def process_submission(input_img, text_answer, country, state, city, se_asia_relevance, culture_knowledge, native_caption, english_caption):
+def process_submission(input_img, text_answer, country, state, city, se_asia_relevance, culture_knowledge, english_caption, native_caption, email):
     # Generate unique ID for this submission
     submission_id = str(uuid.uuid4())
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -97,6 +97,7 @@ def process_submission(input_img, text_answer, country, state, city, se_asia_rel
     submission_data = {
         "id": submission_id,
         "timestamp": timestamp,
+        "email": email,  # Add email as unique identifier
         "image_filename": os.path.basename(image_path) if image_path else None,
         "cultural_relevance": text_answer,
         "country": country,
@@ -104,8 +105,8 @@ def process_submission(input_img, text_answer, country, state, city, se_asia_rel
         "city": city,
         "se_asia_relevance": se_asia_relevance,
         "cultural_knowledge_source": culture_knowledge,
-        "native_caption": native_caption,
-        "english_caption": english_caption
+        "english_caption": english_caption,
+        "native_caption": native_caption
     }
     
     # Save the data as JSON
@@ -141,7 +142,7 @@ def process_submission(input_img, text_answer, country, state, city, se_asia_rel
     
     # Return values to display in the interface
     location_info = f"Location: {city}, {state}, {country}" if state else f"Location: {city}, {country}"
-    return input_img, f"Your text response: {text_answer}", f"Selected location: {location_info}", f"SE Asia relevance: {se_asia_relevance}", f"Cultural knowledge source: {culture_knowledge}", f"Native caption: {native_caption}", f"English caption: {english_caption}"
+    return input_img, f"Your text response: {text_answer}", f"Selected location: {location_info}", f"SE Asia relevance: {se_asia_relevance}", f"Cultural knowledge source: {culture_knowledge}", f"English caption: {english_caption}", f"Native caption: {native_caption}", f"Email: {email}"
 
 # Initialize the dataset
 setup_hf_dataset()
@@ -150,6 +151,9 @@ with gr.Blocks() as gradio_app:
     gr.Markdown("# South Asian Image Data Collection")
     gr.Markdown("Upload an image and answer questions about its cultural significance.")
     
+    with gr.Row():
+        email_input = gr.Textbox(label="Your Email:", placeholder="Enter your email address (will be used as your contributor ID)", info="Your email will be stored with your submission as a unique identifier")
+    
     with gr.Row():
         input_img = gr.Image(label="Upload an image", sources=['upload', 'webcam'], type="pil")
     
@@ -204,26 +208,38 @@ with gr.Blocks() as gradio_app:
     with gr.Row():
         submit_btn = gr.Button("Submit")
     
-    with gr.Row():
-        output_img = gr.Image(label="Submitted Image")
-    
-    with gr.Row():
-        output_text = gr.Text(label="Text Response")
-    
-    with gr.Row():
-        output_location = gr.Text(label="Location Information")
-    
-    with gr.Row():
-        output_relevance = gr.Text(label="South Asia Cultural Relevance")
-    
-    with gr.Row():
-        output_knowledge = gr.Text(label="Cultural Knowledge Source")
-    
-    with gr.Row():
-        output_native = gr.Text(label="Native Language Caption")
+    # Output elements
+    output_elements = gr.Group(visible=False)
+    with output_elements:
+        with gr.Row():
+            output_img = gr.Image(label="Submitted Image")
+        
+        with gr.Row():
+            output_text = gr.Text(label="Text Response")
+        
+        with gr.Row():
+            output_location = gr.Text(label="Location Information")
+        
+        with gr.Row():
+            output_relevance = gr.Text(label="South Asia Cultural Relevance")
+        
+        with gr.Row():
+            output_knowledge = gr.Text(label="Cultural Knowledge Source")
+        
+        with gr.Row():
+            output_english = gr.Text(label="English Caption")
+        
+        with gr.Row():
+            output_native = gr.Text(label="Native Language Caption")
+        
+        with gr.Row():
+            output_email = gr.Text(label="Email")
     
-    with gr.Row():
-        output_english = gr.Text(label="English Caption")
+    # Add function to show outputs after submission
+    def show_outputs(*args):
+        output_values = process_submission(*args)
+        output_elements.visible = True
+        return output_values
     
     # Set up event handlers
     country_dropdown.change(
@@ -233,7 +249,7 @@ with gr.Blocks() as gradio_app:
     )
     
     submit_btn.click(
-        fn=process_submission,
+        fn=show_outputs,
         inputs=[
             input_img,
             text_answer,
@@ -242,8 +258,9 @@ with gr.Blocks() as gradio_app:
             city_textbox,
             se_asia_relevance,
             culture_knowledge,
+            english_caption,
             native_caption,
-            english_caption
+            email_input
         ],
         outputs=[
             output_img,
@@ -251,8 +268,9 @@ with gr.Blocks() as gradio_app:
             output_location,
             output_relevance,
             output_knowledge,
+            output_english,
             output_native,
-            output_english
+            output_email
         ]
     )