Update app.py

app.py

@@ -15,41 +15,38 @@ 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 dictionary
 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"
+        "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", 
+        "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"
+        "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"
+        "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"
+        "Central", "Eastern", "North Central", "Northern", "North Western", "Sabaragamuwa", "Southern", "Uva", "Western"
     ]
 }
 
@@ -72,12 +69,11 @@ def setup_hf_dataset():
     elif not HF_TOKEN:
         print("Warning: HF_TOKEN not set. Data will be stored locally only.")
 
-def update_states(country):
-    """Update the states dropdown based on the selected country"""
+def update_state_dropdown(country):
+    """Update state dropdown based on 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)
+        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):
     # Generate unique ID for this submission
@@ -104,7 +100,7 @@ def process_submission(input_img, text_answer, country, state, city, se_asia_rel
         "image_filename": os.path.basename(image_path) if image_path else None,
         "cultural_relevance": text_answer,
         "country": country,
-        "state_province": state,
+        "state": state,
         "city": city,
         "se_asia_relevance": se_asia_relevance,
         "cultural_knowledge_source": culture_knowledge,
@@ -145,101 +141,118 @@ 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}", 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"Native caption: {native_caption}", f"English caption: {english_caption}"
 
 # Initialize the dataset
 setup_hf_dataset()
 
-with gr.Blocks(title="South Asian Image Data Collection") as gradio_app:
+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():
-        with gr.Column(scale=1):
-            input_img = gr.Image(label="Upload an image", sources=['upload', 'webcam'], type="pil")
-        
-        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"
-            )
+        input_img = gr.Image(label="Upload an image", sources=['upload', 'webcam'], type="pil")
+    
+    with gr.Row():
+        text_answer = gr.Textbox(label="The image portrays culturally-relevant information in:", placeholder="what culture does this image represent?")
+    
+    with gr.Row():
+        country_dropdown = gr.Dropdown(
+            choices=["India", "Pakistan", "Bangladesh", "Afghanistan", "Bhutan", "Nepal", "Sri Lanka"],
+            label="Country where the image was taken:",
+            interactive=True
+        )
+    
+    with gr.Row():
+        state_dropdown = gr.Dropdown(
+            choices=[],
+            label="State/Province:",
+            interactive=True
+        )
+    
+    with gr.Row():
+        city_textbox = gr.Textbox(label="City where the image was taken:", placeholder="Enter city name")
+    
+    with gr.Row():
+        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?"
+        )
+    
+    with gr.Row():
+        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.)"
+        )
+    
+    with gr.Row():
+        native_caption = gr.Textbox(label="Caption in Native Language:", placeholder="Enter caption in the native language of the culture depicted")
+    
+    with gr.Row():
+        english_caption = gr.Textbox(label="English Caption:", placeholder="Enter caption in English")
+    
+    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")
     
-    # Connect the country dropdown to update the state dropdown
+    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")
+    
+    with gr.Row():
+        output_english = gr.Text(label="English Caption")
+    
+    # Set up event handlers
     country_dropdown.change(
-        fn=update_states,
+        fn=update_state_dropdown,
         inputs=country_dropdown,
         outputs=state_dropdown
     )
     
-    submit_btn = gr.Button("Submit")
-    
-    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")
-    
     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
+            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
+            output_img,
+            output_text,
+            output_location,
+            output_relevance,
+            output_knowledge,
+            output_native,
+            output_english
         ]
     )