Update app.py

app.py

@@ -29,7 +29,6 @@ def txt2story(prompt: str, top_k: int, top_p: float, temperature: float) -> str:
         model="meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo",
         messages=[
             {"role": "system", "content": '''As an experienced short story writer, write a meaningful story influenced by the provided prompt.
-            Ensure the story is full of positive inspiration & enthusiasm and concludes with a happy ending. 
             Ensure the story does not exceed 250 words.'''},
             {"role": "user", "content": story_prompt}
         ],
@@ -58,22 +57,11 @@ def txt2speech(text: str) -> None:
     with open('audio_story.mp3', 'wb') as file:
         file.write(response.content)
 
-# Story translation function
-def translate_story(story: str, target_language: str) -> str:
-    # Translation pipeline
-    translation_model = pipeline("translation", model="SnypzZz/Llama2-13b-Language-translate")
-    
-    print(f"Translating the story to {target_language}...")
-    translated_story = translation_model(story, max_length=400, tgt_lang=target_language)[0]['translation_text']
-    
-    return translated_story
-
 # Get user preferences for the story
 def get_user_preferences() -> Dict[str, str]:
     preferences = {}
 
     preferences['continent'] = st.selectbox("Continent", ["North America", "Europe", "Asia", "Africa", "Australia"])
-    preferences['language'] = st.selectbox("Language", ["English", "Hindi", "Spanish", "French", "German"])
     preferences['genre'] = st.selectbox("Genre", ["Science Fiction", "Fantasy", "Mystery", "Romance"])
     preferences['setting'] = st.selectbox("Setting", ["Future", "Medieval times", "Modern day", "Alternate reality"])
     preferences['plot'] = st.selectbox("Plot", ["Hero's journey", "Solving a mystery", "Love story", "Survival"])
@@ -82,7 +70,7 @@ def get_user_preferences() -> Dict[str, str]:
     preferences['conflict'] = st.selectbox("Conflict Type", ["Person vs. Society", "Internal struggle", "Person vs. Nature", "Person vs. Person"])
     preferences['magic_tech'] = st.selectbox("Magic/Technology", ["Advanced technology", "Magic system", "Supernatural abilities", "Alien technology"])
     preferences['twist'] = st.selectbox("Mystery/Twist", ["Plot twist", "Hidden identity", "Unexpected ally/enemy", "Time paradox"])
-    preferences['ending'] = st.selectbox("Ending", ["Bittersweet", "Happy", "Open-ended", "Tragic"])
+    preferences['ending'] = st.selectbox("Ending", ["Happy", "Bittersweet", "Open-ended", "Tragic"])
     
     return preferences
 
@@ -117,7 +105,7 @@ def main():
             scenario = img2txt("uploaded_image.jpg")  # Extracts text from the image
             
             # Modify the prompt to include user preferences
-            prompt = f"Based on the image description: '{scenario}', create a {preferences['genre']} story set in {preferences['setting']}. " \
+            prompt = f"Based on the image description: '{scenario}', create a {preferences['genre']} story set in {preferences['setting']} in {preferences['continent']}. " \
                      f"The story should have a {preferences['tone']} tone and explore the theme of {preferences['theme']}. " \
                      f"The main conflict should be {preferences['conflict']}. " \
                      f"Include {preferences['magic_tech']} as a key element. " \
@@ -125,10 +113,6 @@ def main():
             
             story = txt2story(prompt, top_k, top_p, temperature)  # Generates a story based on the image text, LLM params, and user preferences
             
-            # Translate the story if the user selected a non-English language
-            if preferences['language'] != "English":
-                story = translate_story(story, preferences['language'])
-            
             txt2speech(story)  # Converts the story to audio
 
             st.markdown("---")