update

app.py

@@ -328,8 +328,13 @@ def create_demo():
         # Save story dialogue
         save_btn.click(
             save_story,
-            chatbot,
-            status_msg,
+            inputs=[
+                chatbot,
+                style_select,
+                theme_select, 
+                character_desc
+            ],
+            outputs=status_msg
         )
         
         # User input processing
@@ -427,26 +432,59 @@ def create_demo():
         return demo
 
 
-def save_story(chatbot):
-    """Save story dialogue record"""
+def save_story(chatbot, style=None, theme=None, character_desc=None):
+    """
+    Save story dialogue record with metadata
+    Args:
+        chatbot: Chat history containing user and AI messages
+        style: Story style selected by user
+        theme: Story theme selected by user 
+        character_desc: Character description
+    Returns:
+        Status message indicating success or failure
+    """
     if not chatbot:
         return "Story is empty, cannot save"
         
     timestamp = time.strftime("%Y%m%d_%H%M%S")
-    filename = f"stories/story_{timestamp}.txt"
     
-    os.makedirs("stories", exist_ok=True)
+    # Create stories directory if it doesn't exist
+    # Use absolute path for Hugging Face Space
+    stories_dir = os.path.join(os.getcwd(), "stories")
+    os.makedirs(stories_dir, exist_ok=True)
+    
+    filename = os.path.join(stories_dir, f"story_{timestamp}.txt")
     
     try:
         with open(filename, "w", encoding="utf-8") as f:
-            for user_msg, bot_msg in chatbot:
+            # Write header with metadata
+            f.write("=== Interactive Story ===\n")
+            f.write(f"Created: {time.strftime('%Y-%m-%d %H:%M:%S')}\n")
+            
+            if style:
+                f.write(f"Style: {style}\n")
+            if theme:
+                f.write(f"Theme: {theme}\n")
+            if character_desc:
+                f.write(f"Character: {character_desc}\n")
+            
+            f.write("\n=== Story Content ===\n\n")
+            
+            # Write conversation
+            for i, (user_msg, ai_msg) in enumerate(chatbot, 1):
+                f.write(f"--- Turn {i} ---\n")
                 if user_msg:
                     f.write(f"User: {user_msg}\n")
-                if bot_msg:
-                    f.write(f"AI: {bot_msg}\n\n")
-        return f"Story has been saved to {filename}"
+                if ai_msg:
+                    f.write(f"AI: {ai_msg}\n")
+                f.write("\n")
+                
+        # Return success message with filename
+        return gr.Markdown(f"✅ Story saved successfully to: {os.path.basename(filename)}")
+        
     except Exception as e:
-        return f"Save failed: {str(e)}"
+        logger.error(f"Error saving story: {str(e)}")
+        return gr.Markdown(f"❌ Failed to save story: {str(e)}")
 
 if __name__ == "__main__":
     demo = create_demo()

stories/story_20241212_173910.txt

@@ -0,0 +1,24 @@
+=== Interactive Story ===
+Created: 2024-12-12 17:39:10
+Style: Fantasy
+Theme: Adventure
+Character: A brave and fearless explorer who loves adventure and challenges.
+
+=== Story Content ===
+
+--- Turn 1 ---
+User: A lake deep in the forest
+AI: Deep in the heart of the forest, nestled among towering trees and lush greenery, there was a serene lake that glimmered in the sunlight. It was said that the lake held secrets and treasures beyond imagination, but few dared to venture into its depths. That was until a brave and fearless explorer named Aria set her sights on the lake.
+
+Aria had spent her entire life exploring the unknown and seeking out adventure. Her love for challenges had led her to this very spot, where she stood now, gazing into the shimmering waters. She could feel the excitement building inside of her, and her heart raced with anticipation.
+
+Without hesitation, Aria plunged into the lake, her body slicing through the water with ease. The lake was crystal clear, and she could see the bottom, which was filled with lush plants and colorful fish. As she swam deeper and deeper, Aria's senses heightened. She could hear the gentle lapping of the water against the shore and the rustling of leaves in the wind.
+
+Suddenly, Aria's hand brushed against something cold and hard. Her heart skipped a beat as she reached out and grabbed onto a small object. It was a map, intricately drawn and filled with strange symbols. Aria's eyes widened as she realized the significance of what she had found. This map led to a hidden treasure, and she was the first to discover it.
+
+But Aria knew that she couldn't keep the treasure to herself. It was her duty to share it with the world. With newfound determination, Aria swam back to the shore, the map clutched tightly in her hand. She knew that her adventure was just beginning, and she couldn't wait to see what lay ahead.
+
+As Aria emerged from the lake, she noticed a group of men lurking in the shadows. They were watching her intently, and she knew that they too had set their sights on the treasure. Aria knew that she had to act quickly. She gathered her supplies and set off on a dangerous journey, determined to beat the men to the treasure.
+
+The journey was filled with challenges and obstacles, but Aria was undeterred. She encountered fierce creatures and treacherous terrain, but she pushed on, driven by her love for adventure and the desire to uncover
+