Update app.py

app.py

@@ -27,7 +27,7 @@ def update_state_dropdown(country):
     if country in states_by_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, language, country, state, city, se_asia_relevance, culture_knowledge, native_caption, english_caption,code_mixed_caption,domain,email):
+def process_submission(input_img, language, country, state, city, se_asia_relevance, culture_knowledge, native_caption, english_caption,code_mixed_caption,domain,email,other_language=None):
     submission_id = str(uuid.uuid4())
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     image_path = None
@@ -36,7 +36,9 @@ def process_submission(input_img, language, country, state, city, se_asia_releva
         image_path = os.path.join("uploaded_images", image_filename)
         if isinstance(input_img, str): shutil.copy(input_img, image_path)
         else: input_img.save(image_path)
-    submission_data = {"id": submission_id, "timestamp": timestamp, "image_filename": os.path.basename(image_path) if image_path else None, "cultural_relevance": language, "country": country, "state": state, "city": city, "se_asia_relevance": se_asia_relevance, "cultural_knowledge_source": culture_knowledge, "native_caption": native_caption, "english_caption": english_caption,"code_mixed_caption": code_mixed_caption,"domain":domain, "email": email}
+    submission_data = {"id": submission_id, "timestamp": timestamp, "image_filename": os.path.basename(image_path) if image_path else None, "language": language, "country": country, "state": state, "city": city, "se_asia_relevance": se_asia_relevance, "cultural_knowledge_source": culture_knowledge, "native_caption": native_caption, "english_caption": english_caption,"code_mixed_caption": code_mixed_caption,"domain":domain, "email": email}
+    if other_language is not None:
+        submission_data["other language"] = other_language
     json_filename = f"{timestamp}.json"
     json_path = os.path.join("submissions", json_filename)
     with open(json_path, "w") as f: json.dump(submission_data, f, indent=2)
@@ -49,9 +51,12 @@ def process_submission(input_img, language, country, state, city, se_asia_releva
             print(f"Submission {submission_id} uploaded to Hugging Face Dataset")
         except Exception as e: print(f"Error uploading to dataset: {e}")
     location_info = f"Location: {city}, {state}, {country}" if state else f"Location: {city}, {country}"
-    return input_img, f"Your text response: {language}", 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}" , f"Code mixed caption: {code_mixed_caption}", f"Domain: {domain}"
+    response = [input_img, f"Language: {language}", 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}" , f"Code mixed caption: {code_mixed_caption}", f"Domain: {domain}"]
+    if other_language is not None:
+        response.append(f"Other language: {other_language}")
+    return tuple(response)
 def clear_inputs():
-    return None, "", None, None, "", None, None, "", "", ""
+    return None, "OTHER", "", None, None, "", "", None, None, "", "", "", ""
 setup_hf_dataset()
 with gr.Blocks(theme='1024m/1024m-1') as gradio_app:
     gr.Markdown("# South Asian Image Data Collection")
@@ -60,6 +65,7 @@ with gr.Blocks(theme='1024m/1024m-1') as gradio_app:
         with gr.Column(scale=1):
             input_img = gr.Image(label="Upload an image", sources=['upload', 'webcam'], type="pil")
             language = gr.Dropdown(choices=south_asian_languages, label="Language:", info="Select the native language relevant to the image", interactive=True,value=south_asian_languages[-1])
+            other_language = gr.Textbox(label="Other Language:",placeholder="optional",info="Name of the language if not listed above")
             country_dropdown = gr.Dropdown(choices=["None","India", "Pakistan", "Bangladesh", "Afghanistan", "Bhutan", "Nepal", "Sri Lanka"], label="Country where the image was taken:", interactive=True)
             state_dropdown = gr.Dropdown(choices=[], label="State/Province:", interactive=True)
             city_textbox = gr.Textbox(label="City where the image was taken:", placeholder="Enter city name")
@@ -70,7 +76,7 @@ with gr.Blocks(theme='1024m/1024m-1') as gradio_app:
             native_caption = gr.Textbox(label="Caption in Native Language:", placeholder="Enter caption in the native language of the culture depicted", info="in native script ONLY")
             english_caption = gr.Textbox(label="English Caption:", placeholder="Enter caption in English", info="in english script ONLY")
             code_mixed_caption = gr.Textbox(label="Code-Mixed Caption:", placeholder="Enter caption in code-mixed", info="in english script ONLY")
-            domain = gr.Textbox(label="Domain:",placeholder="description",info="1-2 word any description")
+            domain = gr.Textbox(label="Domain:",placeholder="Description",info="1-2 word any description")
     with gr.Row():
         clear_btn = gr.Button("Clear")
         submit_btn = gr.Button("Submit")
@@ -85,7 +91,7 @@ with gr.Blocks(theme='1024m/1024m-1') as gradio_app:
             output_native = gr.Text(label="Native Language Caption")
             output_english = gr.Text(label="English Caption")
     country_dropdown.change(fn=update_state_dropdown, inputs=country_dropdown, outputs=state_dropdown)
-    submit_btn.click(fn=process_submission, inputs=[input_img, language, country_dropdown, state_dropdown, city_textbox, se_asia_relevance, culture_knowledge, native_caption, english_caption,code_mixed_caption,domain, email_input], outputs=[output_img, output_text, output_location, output_relevance, output_knowledge, output_native, output_english])
-    clear_btn.click(fn=clear_inputs, inputs=[], outputs=[input_img, language, country_dropdown, state_dropdown, city_textbox, se_asia_relevance, culture_knowledge, native_caption, english_caption,code_mixed_caption,domain, email_input])
+    submit_btn.click(fn=process_submission, inputs=[input_img, language, country_dropdown, state_dropdown, city_textbox, se_asia_relevance, culture_knowledge, native_caption, english_caption,code_mixed_caption,domain, email_input,other_language], outputs=[output_img, output_text, output_location, output_relevance, output_knowledge, output_native, output_english])
+    clear_btn.click(fn=clear_inputs, inputs=[], outputs=[input_img, language,other_language, country_dropdown, state_dropdown, city_textbox, se_asia_relevance, culture_knowledge, native_caption, english_caption,code_mixed_caption,domain, email_input])
 if __name__ == "__main__":
     gradio_app.launch()