2024-01-09 11:55:52 +02:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
|
|
import streamlit as st
|
|
|
|
import requests
|
|
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
import os
|
2024-01-09 12:39:15 +02:00
|
|
|
from bidi.algorithm import get_display
|
2024-01-09 11:55:52 +02:00
|
|
|
# Load environment variables from .env file
|
|
|
|
load_dotenv()
|
|
|
|
# Read API key from .env file
|
|
|
|
api_key = os.getenv("API_KEY")
|
|
|
|
|
2024-01-09 12:39:15 +02:00
|
|
|
font = ImageFont.truetype("fonts/BonaNova-Regular.ttf", 40)
|
|
|
|
|
2024-01-09 11:57:06 +02:00
|
|
|
|
2024-01-09 11:55:52 +02:00
|
|
|
def get_nakdan_response(hebrew_text, api_key):
|
|
|
|
url = "https://nakdan-5-3.loadbalancer.dicta.org.il/addnikud"
|
|
|
|
headers = {'Content-Type': 'text/plain;charset=utf-8'}
|
|
|
|
params = {
|
|
|
|
"task": "nakdan",
|
|
|
|
"useTokenization": True,
|
|
|
|
"genre": "modern", # or "rabbinic" or "premodern" based on user's need
|
|
|
|
"data": hebrew_text,
|
|
|
|
"addmorph": True,
|
|
|
|
"matchpartial": True,
|
|
|
|
"keepmetagim": False,
|
|
|
|
"keepqq": False,
|
|
|
|
"apiKey": api_key
|
|
|
|
}
|
|
|
|
response = requests.post(url, headers=headers, json=params)
|
|
|
|
return response.json()
|
|
|
|
|
|
|
|
|
|
|
|
def main():
|
|
|
|
st.title('Dicta Nakdan API Interface')
|
|
|
|
|
|
|
|
# Input fields for user
|
|
|
|
hebrew_text = st.text_area("Enter Hebrew Text:", "טקסט טקסט טקסט")
|
|
|
|
|
|
|
|
# Button to send request
|
|
|
|
if st.button("Process Text"):
|
|
|
|
response = get_nakdan_response(hebrew_text, api_key)
|
2024-01-09 12:39:15 +02:00
|
|
|
st.json(response)
|
2024-01-09 11:55:52 +02:00
|
|
|
# Extract words
|
2024-01-09 11:57:06 +02:00
|
|
|
words = [
|
2024-01-09 12:39:15 +02:00
|
|
|
option['w'].replace("|", "")
|
2024-01-09 11:57:06 +02:00
|
|
|
for item in response['data']
|
|
|
|
if 'nakdan' in item
|
|
|
|
for option in item['nakdan'].get('options', [])
|
|
|
|
]
|
2024-01-09 11:55:52 +02:00
|
|
|
st.text(words)
|
|
|
|
# Create an image
|
2024-01-09 12:39:15 +02:00
|
|
|
img = Image.new('RGB', (300, 400), color=(255, 255, 255))
|
2024-01-09 11:55:52 +02:00
|
|
|
d = ImageDraw.Draw(img)
|
|
|
|
|
|
|
|
# Position for the first word
|
|
|
|
x, y = 10, 40
|
|
|
|
|
|
|
|
# Write words on the image
|
|
|
|
for word in words:
|
2024-01-09 12:39:15 +02:00
|
|
|
bidi_word = get_display(word) # Convert word to RTL format
|
|
|
|
d.text((x, y), bidi_word, fill=(0, 0, 0), font=font)
|
|
|
|
y += 45 # Move to the next line
|
2024-01-09 11:55:52 +02:00
|
|
|
st.image(img)
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
2024-01-09 12:39:15 +02:00
|
|
|
main()
|