Update app.py

app.py

@@ -66,4 +66,134 @@ def generate_poem(selected_language, text):
             image = generate_image_from_poem(translated_text)  # Return the image from the generate_image_from_poem function
 
         return poem, (sampling_rate, audio_data), image
-    except Exc
+    except Exception as e:
+        return f"Error: {str(e)}", None, None
+
+"""### **Poem Generation Function:**
+This function is responsible for generating a poem (text) in Arabic or English, based on the provided text.
+"""
+
+# Poem generation for Arabic
+def generate_poem_arabic(text):
+    generated_text = pipe_ar(text, max_length=96, do_sample=True, temperature=1.0, top_k=50, top_p=0.9,
+                             repetition_penalty=1.2, min_length=64, no_repeat_ngram_size=3, num_beams=5,
+                             num_return_sequences=1)[0]["generated_text"]
+    clean_text = generated_text.replace("-", "")  # To get rid of the dashes generated by the model.
+    return clean_text
+
+# Poem generation for English using AutoTokenizer and AutoModelForCausalLM
+def generate_poem_english(text):
+    inputs = tokenizer_en(text, return_tensors="pt")
+    outputs = model_en.generate(**inputs, max_length=100, do_sample=True, top_k=50, top_p=0.9, temperature=1.0)
+    generated_text = tokenizer_en.decode(outputs[0], skip_special_tokens=True)
+    clean_text = generated_text.replace("</s>", "")  # Clean any unwanted tokens
+    return clean_text
+
+"""### **Audio Function:**
+This function is responsible for generating audio in Arabic or English, based on the provided text.
+"""
+
+# Text-to-speech conversion for Arabic
+def text_to_speech_arabic(text):
+    speech = synthesiser_arabic(text, forward_params={"speaker_embeddings": speaker_embedding_arabic})
+    audio_data = speech["audio"]
+    sampling_rate = speech["sampling_rate"]
+    return (sampling_rate, audio_data)
+
+# Text-to-speech conversion for English
+def text_to_speech_english(text):
+    speech = synthesiser_english(text, forward_params={"speaker_embeddings": speaker_embedding_english})
+    audio_data = speech["audio"]
+    sampling_rate = speech["sampling_rate"]
+    return (sampling_rate, audio_data)
+
+"""### **Image Function:**
+This function is responsible for generating an image based on the provided text.
+"""
+
+# Image generation function
+def generate_image_from_poem(poem_text):
+    image = pipe_image(poem_text).images[0]
+    return image
+
+"""### **Translation Function:**
+This function is responsible for translating Arabic input to English, to be used for the image function, which accepts only English inputs.
+"""
+
+# Translation function from Arabic to English
+def translate_arabic_to_english(text):
+    translated_text = pipe_translator(text)[0]['translation_text']
+    return translated_text
+
+"""### **CSS Styling:**"""
+
+custom_css = """
+body {
+    background-color: #f4f4f9;
+    color: #333;
+}
+.gradio-container {
+    border-radius: 10px;
+    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
+    background-color: #fff;
+}
+label {
+    color: #4A90E2;
+    font-weight: bold;
+}
+
+input[type="text"],
+textarea {
+    border: 1px solid #4A90E2;
+}
+textarea {
+    height: 150px;
+}
+
+button {
+    background-color: #4A90E2;
+    color: #fff;
+    border-radius: 5px;
+    cursor: pointer;
+}
+button:hover {
+    background-color: #357ABD;
+}
+
+.dropdown {
+    border: 1px solid #4A90E2;
+    border-radius: 4px;
+}
+
+"""
+
+"""### **Examples for Gradio:**
+Provide 4 predefined inputs to demonstrate how the interface works.
+"""
+
+examples = [
+    ["English", "The shining sun rises over the calm ocean"],
+    ["Arabic", "الورود تتفتح في الربيع"],
+    ["English", "The night sky is filled with stars and dreams"],
+    ["Arabic", "اشعة الشمس المشرقة"]
+]
+
+"""### **Gradio Interface:**
+Creating a Gradio interface to generate a poem, read the poem, and generate an image based on that poem.
+"""
+
+my_model = gr.Interface(
+    fn=generate_poem,  # The primary function that will receive the inputs (language and the starter of the poem)
+    inputs=[
+        gr.Dropdown(["English", "Arabic"], label="Select Language"),  # Dropdown menu to select the language
+        gr.Textbox(label="Enter a sentence")  # Textbox where the user will input a sentence or phrase to generate the poem
+    ],
+    outputs=[
+        gr.Textbox(label="Generated Poem", lines=10),  # Textbox to display the generated poem
+        gr.Audio(label="Generated Audio", type="numpy"),  # Audio output for the generated poem
+        gr.Image(label="Generated Image")  # Image output for the generated image
+    ],
+    examples=examples,  # Predefined examples to guide the user
+    css=custom_css  # Applying custom CSS
+)
+my_model.launch()