Jesivn's picture
Update app.py
74f1ddf verified
raw
history blame
6.61 kB
import streamlit as st
from transformers import pipeline
import torch
import time
import requests
import io
import os
from PIL import Image
# Load models
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-dra-en")
# for summarizer api
SUMMARIZER_API_URL = "https://api.groq.com/openai/v1/chat/completions"
summarizer_headers = {"Authorization": f"Bearer {os.getenv('GROQ_API_TOKEN')}",
"Content-Type": "application/json"}
# for image api
IMAGE_API_URL = "https://api-inference.huggingface.co/models/ZB-Tech/Text-to-Image"
img_headers = {"Authorization": f"Bearer {os.getenv('HF_API_TOKEN')}"}
# Functions for each task
def translate_tamil_to_english(text):
time.sleep(2)
result = translator(text)
return result[0]['translation_text']
def summarize_english_text(paragraph):
time.sleep(2)
# Request payload
payload = {
"model": "mixtral-8x7b-32768",
"messages": [
{"role": "system", "content": "Create a summary of below paragraph in 30 words max"},
{"role": "user", "content": paragraph}
],
"max_tokens": 100 # number of words in the output.
}
# Send POST request to Groq API
response = requests.post(SUMMARIZER_API_URL, json=payload, headers=summarizer_headers)
# Check if the request was successful
if response.status_code == 200:
# Parse the JSON response
result = response.json()
# Extract and print the generated text
generated_text = result['choices'][0]['message']['content']
return generated_text
else:
return f"Error: {response.status_code}, {response.text}"
def english_text_to_image(text):
payload = {
"inputs": prompt,
}
response = requests.post(IMAGE_API_URL, headers=img_headers, json=payload)
image_bytes = response.content
image = Image.open(io.BytesIO(image_bytes))
return image
# Custom CSS
st.markdown("""
<style>
/* Background color */
body {
background-color: #f0f0f5;
}
/* Text color and font */
.stApp {
font-family: 'Arial', sans-serif;
color: #333;
}
/* Titles and subtitles styling */
h1 {
color: #2E8B57;
text-align: center;
text-shadow: 2px 2px 5px #aaaaaa;
}
h2, h3 {
color: #4682B4;
text-shadow: 1px 1px 3px #aaaaaa;
}
/* Background texture */
.stApp {
background: linear-gradient(to bottom right, #fff7e6, #e6f7ff);
}
/* Button styling */
button[kind="primary"] {
background-color: #4682B4;
color: white;
border-radius: 8px;
padding: 0.5rem 1rem;
}
button[kind="primary"]:hover {
background-color: #5b9bd5;
}
/* Text area and input field styling */
textarea, input {
border-radius: 10px;
padding: 1rem;
border: 2px solid #ccc;
background-color: #f9f9f9;
}
/* Styling the output boxes */
.stMarkdown {
background-color: #e6f9ff;
padding: 1rem;
border-radius: 10px;
box-shadow: 2px 2px 10px #ccc;
}
</style>
""", unsafe_allow_html=True)
#sidebar styling
st.markdown("""
<style>
[data-testid=stSidebar] {
background-color: #FFFFFF;
margin-right: 20px;
border-right: 2px solid #FFFFFF
}
</style>
""", unsafe_allow_html=True)
#options styling in sidebar and added image in sidebar
with st.sidebar:
selected = option_menu(
menu_title="",
options=['Home','Tool'],
icons=['house-door-fill','setting'],
menu_icon='truck-front-fill',
default_index=0,
styles={
"container": {'padding':'5!important','background-color':'#FAF9F6'},
"icon": {'color':"#000000", "font-size":"23px"},
"nav-link": {'font-size':'16px','text-align':'left','margin':'0px','--hover-color':'#EDEADE','font-weight':'bold'},
"nav-link-selector":{'background-color':'#E6E6FA','font-weight':'bold'}
}
)
if selected == "Home":
# Page title and header
st.title(":blue[Multi-Purpose Tool] - Empowering Educators πŸŽ“")
# Subheader for the app description
st.subheader("A versatile tool designed to assist teachers in translating, summarizing, and visualizing concepts.")
# Main description with detailed information about the app
st.markdown("""
The **Multi-Purpose Tool** is a user-friendly platform developed for educators,
enabling them to enhance their teaching experience. Whether it's translating content
into different languages, summarizing lengthy materials, or visualizing concepts
through images, this tool provides a one-stop solution for modern teaching needs.
### Key Features:
- **Translation**: Translate text seamlessly between languages (e.g., Tamil to English).
- **Summarization**: Quickly generate summaries of long passages for easy understanding.
- **Text to Image**: Visualize difficult concepts by generating images from text descriptions.
### Available Worldwide:
The Multi-Purpose Tool is deployed on Hugging Face and accessible globally to teachers
and educators at the click of a button. Visit the [app here](https://huggingface.co/spaces/Jesivn/Multi_Purpose_Tool).
Empower your classroom with advanced AI tools today!
""")
elif selected=="Tool":
# Row 1: Tamil to English translation
st.subheader("🌐 Translate Tamil to English")
tamil_input = st.text_area("Enter Tamil text", "")
if st.button("Translate"):
english_output = translate_tamil_to_english(tamil_input)
st.markdown(f"**Translated English Text**: {english_output}")
# Row 2: English paragraph summarization
st.subheader("πŸ“ Summarize English Paragraph")
english_paragraph = st.text_area("Enter English paragraph", "")
if st.button("Summarize"):
summary_output = summarize_english_text(english_paragraph)
st.markdown(f"**Summary**: {summary_output}")
# Row 3: English text to image generation
st.subheader("🎨 Generate Image from English Text")
image_text = st.text_input("Enter description for image generation", "")
if st.button("Generate Image"):
generated_image = english_text_to_image(image_text)
st.image(generated_image, caption="Generated Image")