Update app.py

app.py

@@ -15,6 +15,44 @@ HF_TOKEN = os.environ.get("HF_TOKEN")  # You'll need to set this as a secret in
 DATASET_NAME = "srishtiy/se-culture-dataset-results"  # Change to your username/dataset-name
 DATASET_CREATED = False
 
+# Define states/provinces for each country
+states_by_country = {
+    "India": [
+        "Andhra Pradesh", "Arunachal Pradesh", "Assam", "Bihar", "Chhattisgarh", "Goa", 
+        "Gujarat", "Haryana", "Himachal Pradesh", "Jharkhand", "Karnataka", "Kerala", 
+        "Madhya Pradesh", "Maharashtra", "Manipur", "Meghalaya", "Mizoram", "Nagaland", 
+        "Odisha", "Punjab", "Rajasthan", "Sikkim", "Tamil Nadu", "Telangana", "Tripura", 
+        "Uttar Pradesh", "Uttarakhand", "West Bengal", "Andaman and Nicobar Islands", 
+        "Chandigarh", "Dadra and Nagar Haveli and Daman and Diu", "Delhi", "Jammu and Kashmir", 
+        "Ladakh", "Lakshadweep", "Puducherry"
+    ],
+    "Pakistan": [
+        "Balochistan", "Khyber Pakhtunkhwa", "Punjab", "Sindh", "Islamabad Capital Territory", 
+        "Azad Jammu and Kashmir", "Gilgit-Baltistan"
+    ],
+    "Bangladesh": [
+        "Barisal", "Chittagong", "Dhaka", "Khulna", "Mymensingh", "Rajshahi", "Rangpur", "Sylhet"
+    ],
+    "Afghanistan": [
+        "Badakhshan", "Badghis", "Baghlan", "Balkh", "Bamyan", "Daykundi", "Farah", "Faryab", 
+        "Ghazni", "Ghor", "Helmand", "Herat", "Jowzjan", "Kabul", "Kandahar", "Kapisa", "Khost", 
+        "Kunar", "Kunduz", "Laghman", "Logar", "Nangarhar", "Nimruz", "Nuristan", "Paktia", 
+        "Paktika", "Panjshir", "Parwan", "Samangan", "Sar-e Pol", "Takhar", "Uruzgan", "Wardak", "Zabul"
+    ],
+    "Bhutan": [
+        "Bumthang", "Chukha", "Dagana", "Gasa", "Haa", "Lhuentse", "Mongar", "Paro", "Pemagatshel", 
+        "Punakha", "Samdrup Jongkhar", "Samtse", "Sarpang", "Thimphu", "Trashigang", "Trashiyangtse", 
+        "Trongsa", "Tsirang", "Wangdue Phodrang", "Zhemgang"
+    ],
+    "Nepal": [
+        "Bagmati", "Gandaki", "Karnali", "Koshi", "Lumbini", "Madhesh", "Sudurpashchim"
+    ],
+    "Sri Lanka": [
+        "Central", "Eastern", "North Central", "Northern", "North Western", "Sabaragamuwa", 
+        "Southern", "Uva", "Western"
+    ]
+}
+
 def setup_hf_dataset():
     """Initialize the Hugging Face dataset if it doesn't exist"""
     global DATASET_CREATED
@@ -34,7 +72,14 @@ def setup_hf_dataset():
     elif not HF_TOKEN:
         print("Warning: HF_TOKEN not set. Data will be stored locally only.")
 
-def process_submission(input_img, text_answer, multiple_choice, city, country, se_asia_relevance, culture_knowledge, native_caption, english_caption):
+def update_states(country):
+    """Update the states dropdown based on the selected country"""
+    if country in states_by_country:
+        return gr.Dropdown.update(choices=states_by_country[country], visible=True)
+    else:
+        return gr.Dropdown.update(choices=[], value=None, visible=False)
+
+def process_submission(input_img, text_answer, country, state, city, se_asia_relevance, culture_knowledge, native_caption, english_caption):
     # Generate unique ID for this submission
     submission_id = str(uuid.uuid4())
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -58,9 +103,9 @@ def process_submission(input_img, text_answer, multiple_choice, city, country, s
         "timestamp": timestamp,
         "image_filename": os.path.basename(image_path) if image_path else None,
         "cultural_relevance": text_answer,
-        "continent": multiple_choice,
-        "city": city,
         "country": country,
+        "state_province": state,
+        "city": city,
         "se_asia_relevance": se_asia_relevance,
         "cultural_knowledge_source": culture_knowledge,
         "native_caption": native_caption,
@@ -99,102 +144,104 @@ def process_submission(input_img, text_answer, multiple_choice, city, country, s
             print(f"Error uploading to dataset: {e}")
     
     # Return values to display in the interface
-    return input_img, f"Your text response: {text_answer}", f"Your selected option: {multiple_choice}", f"Location: {city}, {country}", f"SE Asia relevance: {se_asia_relevance}", f"Cultural knowledge source: {culture_knowledge}", f"Native caption: {native_caption}", f"English caption: {english_caption}"
+    location_info = f"Location: {city}, {state}, {country}" if state else f"Location: {city}, {country}"
+    return input_img, f"Your text response: {text_answer}", 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}"
 
 # Initialize the dataset
 setup_hf_dataset()
 
-gradio_app = gr.Interface(
-    process_submission,
-    with gr.Blocks() as gradio_app:
-        selected_country = gr.Dropdown(
-            choices=list(country_to_states.keys()),
-            label="Country where the image was taken:",
-            interactive=True
-        )
-        selected_state = gr.Dropdown(
-            choices=[],
-            label="State / Province / District (updates based on country)",
-            interactive=True
-        )
-        city_field = gr.Textbox(label="City (optional)")
+with gr.Blocks(title="South Asian Image Data Collection") as gradio_app:
+    gr.Markdown("# South Asian Image Data Collection")
+    gr.Markdown("Upload an image and answer questions about its cultural significance.")
     
-        image_input = gr.Image(label="Upload an image", sources=['upload', 'webcam'], type="pil")
-        cultural_text = gr.Textbox(label="The image portrays culturally-relevant information in:")
+    with gr.Row():
+        with gr.Column(scale=1):
+            input_img = gr.Image(label="Upload an image", sources=['upload', 'webcam'], type="pil")
         
-        sa_relevance = gr.Radio(
-            choices=[
-                "Yes. Unique to South Asia",
-                "Yes, people will likely think of South Asia when seeing the picture, but it may have low degree of similarity to other cultures.",
-                "Maybe, this culture did not originate from South Asia, but it's quite dominant in South Asia",
-                "Not really. It has some affiliation to South Asia, but actually does not represent South Asia or has stronger affiliation to cultures outside South Asia",
-                "No. Totally unrelated to South Asia"
-            ],
-            label="Is the image culturally relevant in South Asia?"
-        )
-    
-        knowledge_source = gr.Radio(
-            choices=[
-                "I'm from this country/culture",
-                "I checked online resources (e.g., Wikipedia, articles, blogs)"
-            ],
-            label="How do you know about this culture?",
-            info="Please do not consult LLMs"
-        )
-    
-        native_caption = gr.Textbox(label="Caption in Native Language:")
-        english_caption = gr.Textbox(label="English Caption:")
-    
-        output_components = [
-            gr.Image(label="Submitted Image"),
-            gr.Text(label="Text Response"),
-            gr.Text(label="Country and State"),
-            gr.Text(label="Location Information"),
-            gr.Text(label="South Asia Cultural Relevance"),
-            gr.Text(label="Cultural Knowledge Source"),
-            gr.Text(label="Native Language Caption"),
-            gr.Text(label="English Caption")
-        ]
-    
-        # Link dynamic dropdown update
-        def update_states(selected_country):
-            return gr.Dropdown.update(choices=country_to_states.get(selected_country, []), value=None)
+        with gr.Column(scale=1):
+            text_answer = gr.Textbox(
+                label="The image portrays culturally-relevant information in:", 
+                placeholder="What culture does this image represent?"
+            )
+            
+            # Location information with dynamic dropdowns
+            country_dropdown = gr.Dropdown(
+                choices=list(states_by_country.keys()),
+                label="Country where the image was taken:",
+                interactive=True
+            )
+            
+            state_dropdown = gr.Dropdown(
+                label="State/Province/District:",
+                interactive=True,
+                visible=False  # Will be made visible when country is selected
+            )
+            
+            city_textbox = gr.Textbox(
+                label="City (optional):", 
+                placeholder="Enter city name"
+            )
+            
+            se_asia_relevance = gr.Radio(
+                choices=[
+                    "Yes. Unique to South Asia",
+                    "Yes, people will likely think of South Asia when seeing the picture, but it may have low degree of similarity to other cultures.",
+                    "Maybe, this culture did not originate from South Asia, but it's quite dominant in South Asia",
+                    "Not really. It has some affiliation to South Asia, but actually does not represent South Asia or has stronger affiliation to cultures outside South Asia",
+                    "No. Totally unrelated to South Asia"
+                ],
+                label="Is the image culturally relevant in South Asia?"
+            )
+            
+            culture_knowledge = gr.Radio(
+                choices=[
+                    "I'm from this country/culture",
+                    "I checked online resources (e.g., Wikipedia, articles, blogs)"
+                ],
+                label="How do you know about this culture?",
+                info="Please do not consult LLMs (e.g., GPT-4o, Claude, Command-R, etc.)"
+            )
+            
+            native_caption = gr.Textbox(
+                label="Caption in Native Language:", 
+                placeholder="Enter caption in the native language of the culture depicted"
+            )
+            
+            english_caption = gr.Textbox(
+                label="English Caption:", 
+                placeholder="Enter caption in English"
+            )
     
-        selected_country.change(fn=update_states, inputs=selected_country, outputs=selected_state)
+    # Connect the country dropdown to update the state dropdown
+    country_dropdown.change(
+        fn=update_states,
+        inputs=country_dropdown,
+        outputs=state_dropdown
+    )
     
-        submit_btn = gr.Button("Submit")
+    submit_btn = gr.Button("Submit")
     
-        submit_btn.click(
-            process_submission,
-            inputs=[
-                image_input,
-                cultural_text,
-                selected_country,
-                selected_state,
-                city_field,
-                sa_relevance,
-                knowledge_source,
-                native_caption,
-                english_caption
-            ],
-            outputs=output_components
-        )
+    with gr.Row():
+        with gr.Column():
+            output_img = gr.Image(label="Submitted Image")
+            text_response = gr.Text(label="Text Response")
+            location_info = gr.Text(label="Location Information")
+            relevance_output = gr.Text(label="South Asia Cultural Relevance")
+            knowledge_source = gr.Text(label="Cultural Knowledge Source")
+            native_caption_output = gr.Text(label="Native Language Caption")
+            english_caption_output = gr.Text(label="English Caption")
     
-        gr.Markdown("## South Asian Image Data Collection")
-        gr.Markdown("Upload an image and answer questions about its cultural significance."),
-    outputs=[
-        gr.Image(label="Submitted Image"),
-        gr.Text(label="Text Response"),
-        gr.Text(label="Multiple Choice Response"),
-        gr.Text(label="Location Information"),
-        gr.Text(label="South Asia Cultural Relevance"),
-        gr.Text(label="Cultural Knowledge Source"),
-        gr.Text(label="Native Language Caption"),
-        gr.Text(label="English Caption")
-    ],
-    title="South Asian Image Data Collection",
-    description="Upload an image and answer questions about its cultural significance."
-)
+    submit_btn.click(
+        fn=process_submission,
+        inputs=[
+            input_img, text_answer, country_dropdown, state_dropdown, city_textbox,
+            se_asia_relevance, culture_knowledge, native_caption, english_caption
+        ],
+        outputs=[
+            output_img, text_response, location_info, relevance_output,
+            knowledge_source, native_caption_output, english_caption_output
+        ]
+    )
 
 if __name__ == "__main__":
     gradio_app.launch()