Update backup2.history.app.py

backup2.history.app.py

@@ -15,13 +15,18 @@ from gradio_client import Client
 
 warnings.filterwarnings('ignore')
 
-# Initialize story starters
+# Initialize story starters with added comedy section
 STORY_STARTERS = [
     ['Adventure', 'In a hidden temple deep in the Amazon...'],
     ['Mystery', 'The detective found an unusual note...'],
     ['Romance', 'Two strangers meet on a rainy evening...'],
     ['Sci-Fi', 'The space station received an unexpected signal...'],
-    ['Fantasy', 'A magical portal appeared in the garden...']
+    ['Fantasy', 'A magical portal appeared in the garden...'],
+    ['Comedy-Sitcom', 'The new roommate arrived with seven emotional support animals...'],
+    ['Comedy-Workplace', 'The office printer started sending mysterious messages...'],
+    ['Comedy-Family', 'Grandma decided to become a social media influencer...'],
+    ['Comedy-Supernatural', 'The ghost haunting the house was absolutely terrible at scaring people...'],
+    ['Comedy-Travel', 'The GPS insisted on giving directions in interpretive dance descriptions...']
 ]
 
 # Initialize client outside of interface definition
@@ -34,58 +39,70 @@ def init_client():
     return arxiv_client
 
 def save_story(story, audio_path):
-    """Save story and audio to gallery"""
+    """Save story and audio to gallery with markdown formatting"""
     try:
         # Create gallery directory if it doesn't exist
         gallery_dir = Path("gallery")
         gallery_dir.mkdir(exist_ok=True)
         
-        # Generate timestamp for unique filename
+        # Generate timestamp and sanitize first line for filename
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        first_line = story.split('\n')[0].strip()
+        safe_name = re.sub(r'[^\w\s-]', '', first_line)[:50]  # First 50 chars, sanitized
         
-        # Save story text
-        story_path = gallery_dir / f"story_{timestamp}.txt"
+        # Save story text as markdown
+        story_path = gallery_dir / f"story_{timestamp}_{safe_name}.md"
         with open(story_path, "w") as f:
-            f.write(story)
+            f.write(f"# {first_line}\n\n{story}")
         
-        # Copy audio file to gallery
+        # Copy audio file to gallery with matching name
+        new_audio_path = None
         if audio_path:
-            new_audio_path = gallery_dir / f"audio_{timestamp}.mp3"
+            new_audio_path = gallery_dir / f"audio_{timestamp}_{safe_name}.mp3"
             os.system(f"cp {audio_path} {str(new_audio_path)}")
             
-        return str(story_path), str(new_audio_path)
+        return str(story_path), str(new_audio_path) if new_audio_path else None
     except Exception as e:
         print(f"Error saving to gallery: {str(e)}")
         return None, None
 
 def load_gallery():
-    """Load all stories and audio from gallery"""
+    """Load all stories and audio from gallery with markdown support"""
     try:
         gallery_dir = Path("gallery")
         if not gallery_dir.exists():
             return []
         
         files = []
-        for story_file in gallery_dir.glob("story_*.txt"):
-            timestamp = story_file.stem.split('_')[1]
-            audio_file = gallery_dir / f"audio_{timestamp}.mp3"
+        for story_file in sorted(gallery_dir.glob("story_*.md"), reverse=True):
+            # Extract timestamp and name from filename
+            parts = story_file.stem.split('_', 2)
+            timestamp = f"{parts[1]}"
             
+            # Find matching audio file
+            audio_pattern = f"audio_{timestamp}_*.mp3"
+            audio_files = list(gallery_dir.glob(audio_pattern))
+            audio_file = audio_files[0] if audio_files else None
+            
+            # Read story content and get preview
             with open(story_file) as f:
-                story_text = f.read()
+                content = f.read()
+                # Skip markdown header and get preview
+                preview = content.split('\n\n', 1)[1][:100] + "..."
             
-            # Format as list instead of dict for Gradio Dataframe
             files.append([
                 timestamp,
-                story_text[:100] + "...",
+                f"[{preview}]({str(story_file)})",  # Markdown link to story
                 str(story_file),
-                str(audio_file) if audio_file.exists() else None
+                str(audio_file) if audio_file else None
             ])
         
-        return sorted(files, key=lambda x: x[0], reverse=True)
+        return files
     except Exception as e:
         print(f"Error loading gallery: {str(e)}")
         return []
 
+# Keep all other functions unchanged
 def generate_story(prompt, model_choice):
     """Generate story using specified model"""
     try:
@@ -145,7 +162,7 @@ def play_gallery_audio(evt: gr.SelectData, gallery_data):
         print(f"Error playing gallery audio: {str(e)}")
         return None
 
-# Create the Gradio interface
+# Create the Gradio interface (keep unchanged)
 with gr.Blocks(title="AI Story Generator") as demo:
     gr.Markdown("""
     # 🎭 AI Story Generator & Narrator