thak123's picture
Update app.py
f3091e4
raw
history blame
3.92 kB
import gradio as gr
import random
import nltk
from nltk.corpus import wordnet
from PIL import Image
import requests
# Download the Croatian WordNet data
nltk.download("wordnet")
nltk.download("omw")
nltk.download("omw-1.4")
# Function to get the English WordNet synset ID for a Croatian word
def get_english_synset_id(croatian_word):
# Convert the Croatian word to English WordNet synset IDs
synsets = wordnet.synsets(croatian_word, lang="hrv")
print(f"synsets: {synsets}")
# Check if any synsets were found
if synsets:
# Take the first synset (you can modify this to handle multiple synsets if needed)
hrv_synset = synsets[0]
english_synset_id = wordnet.synset_from_pos_and_offset("n", hrv_synset.offset())
return english_synset_id
else:
return None
# Function to get definitions and images from web
def get_definition_and_image(word):
# try:
# # You may need to adjust this URL based on the source of definitions and images
# url = f"https://your-dictionary-website.com/{word}"
# response = requests.get(url)
# soup = BeautifulSoup(response.text, "html.parser")
# # Extract definition and image (assuming they are in specific HTML elements)
# definition = soup.find("div", {"class": "definition"}).text
# image_url = soup.find("img", {"class": "word-image"})["src"]
# return definition, image_url
# except Exception as e:
# return None, None
# Define the WordNet synset ID for the concept you're interested in
synset_id = get_english_synset_id(word) #"n07753592" # Replace with the synset ID you want
# URL for the ImageNet API
base_url = "http://www.image-net.org/api/text/imagenet.synset.geturls?wnid="
if not synset_id: return None, None
# Cofrom PIL import Image mbine the base URL and the synset ID to get a list of image URLs
url = base_url + str(synset_id.offset())
# Send a GET request to the URL
response = requests.get(url)
if response.status_code == 200:
# Parse the response to get a list of image URLs
image_urls = response.text.splitlines()
# Print the first 10 image URLs
for i, url in enumerate(image_urls[:10], start=1):
print(f"Image {i}: {url}")
return word + ":" + synset_id.definition() or "" , image_urls[0]
else:
print(f"Failed to retrieve image URLs. Status code: {response.status_code}")
return None, None
# Function to randomly sample a noun or verb
def sample_word_and_display_info(dummy="dummy"):
is_noun = random.choice([True, False]) # Randomly choose noun or verb
if is_noun:
pos = "n"
else:
pos = "v"
# Get a random word of the chosen part of speech from the Croatian WordNet
word = random.choice(list(wordnet.all_lemma_names(pos, lang="hrv")))
print(f"Word: {word}")
# Get the definition and image (change the function for your source)
definition, image_url = get_definition_and_image(word)
if definition and image_url:
print(f"Word: {word}")
print(f"Part of speech: {'Noun' if is_noun else 'Verb'}")
print(f"Definition: {definition}")
print(f"Image URL: {image_url}")
return image_url ,definition
else:
print(f"No information found for '{word}'")
img = Image.new('RGB', (64,64) )
return img, f"No information found for '{word}'"
with gr.Blocks() as demo:
image, text = sample_word_and_display_info()
print(image, text)
with gr.Row():
img_1= gr.Image(label="image")
inp = gr.Textbox(placeholder="What is your name?")
text_1 = gr.Markdown(label="text")
button = gr.Button(label="Update Text")
button.click(sample_word_and_display_info,inputs=inp, outputs=text_1)
demo.launch()