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,email):
+def process_submission(input_img, language, country, state, city, se_asia_relevance, culture_knowledge, native_caption, english_caption,code_mixed_caption,domain,email):
     submission_id = str(uuid.uuid4())
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     image_path = None
@@ -36,7 +36,7 @@ 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, "email": email}
+    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}
     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,7 +49,7 @@ 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}"
+    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}"
 def clear_inputs():
     return None, "", None, None, "", None, None, "", "", ""
 setup_hf_dataset()
@@ -59,7 +59,7 @@ with gr.Blocks(theme='1024m/1024m-1') as gradio_app:
     with gr.Row():
         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)
+            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])
             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")
@@ -67,8 +67,10 @@ with gr.Blocks(theme='1024m/1024m-1') as gradio_app:
         with gr.Column(scale=1):
             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")
+            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")
     with gr.Row():
         clear_btn = gr.Button("Clear")
         submit_btn = gr.Button("Submit")
@@ -83,7 +85,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, 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, 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], 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])
 if __name__ == "__main__":
     gradio_app.launch()