fixed interface.py

interface.py

@@ -1,13 +1,14 @@
 import streamlit as st
+from yume import Yume
+from yume.config import yume_small
 
 
 def main():
-    st.set_page_config(page_title="夢を見ている", page_icon="🌙", layout="wide")
+    st.set_page_config(page_title="夢の生成器", page_icon="🌙", layout="wide")
 
-    st.title("夢を見ている")
-    st.write("Welcome to our Dream Writer! 🌙")
+    st.title("夢の生成器")
+    st.write("Welcome to our Dream Generator! 🌙")
 
-    # Add a sidebar with options
     st.sidebar.header("Select a Dream Theme")
     theme = st.sidebar.selectbox(
         "Choose a theme", ["Fantasy", "Adventure", "Romance", "Horror"]
@@ -16,32 +17,33 @@ def main():
     st.sidebar.header("Customize Your Dream")
     prompt = st.sidebar.text_input("Enter a prompt or keyword")
     num_sentences = st.sidebar.slider("Number of sentences to generate", 1, 10, 5)
-    temperature = st.sidebar.slider("Creativity of LLM (Temperature)", 1, 10, 5)
+    temperature = st.sidebar.slider("Creativity of LLM (Temperature)", 0.1, 1.0, 0.5)
 
     generate_button = st.sidebar.button("Generate Dream")
 
-    # Create a main area to display the generated text
     st.header("Your Dream")
 
     if generate_button:
-        # Add a loading animation
+
         with st.spinner("Generating dream..."):
-            # Generate text using the LLM
-            # (assuming you have an LLM model implemented)
-            dream_text = generate_dream_text(theme, prompt, num_sentences)
 
-            # Display the generated text
+            dream_text = generate_dream_text(theme, prompt, num_sentences, temperature)
+
             st.text_area("", value=dream_text, height=600)
     else:
-        text_area = st.text_area(
-            "", height=600, placeholder="Your dream will appear here..."
-        )
+        st.text_area("", height=600, placeholder="Your dream will appear here...")
+
+
+def generate_dream_text(theme, prompt, num_sentences, temperature):
+
+    yume = Yume(config=yume_small)
+
+    yume.load_pretrained("zaibutcooler/yume")
 
+    full_prompt = f"{theme} - {prompt}"
 
-def generate_dream_text(theme, prompt, num_sentences):
-    # Implement your LLM model here to generate text
-    # For demonstration purposes, return a dummy text
-    return "This is a dummy dream text. Please implement your LLM model to generate actual text."
+    # Generate text using the Yume model
+    return yume.sample(full_prompt)
 
 
 if __name__ == "__main__":