Update app.py

app.py

@@ -1,4 +1,3 @@
-# Keep all existing imports and add any new ones needed
 import gradio as gr
 import random
 from datetime import datetime
@@ -16,13 +15,62 @@ from gradio_client import Client
 
 warnings.filterwarnings('ignore')
 
-# Add constants for community generations
+# Initialize constants
 PAGE_SIZE = 10
-FILE_DIR_PATH = "gallery"  # Using existing gallery directory
+FILE_DIR_PATH = "gallery"
 
-# Keep existing STORY_STARTERS array
+# Initialize story starters
+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...'],
+    ['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...']
+]
 
-# Keep all existing functions through play_gallery_audio()
+# Initialize client outside of interface definition
+arxiv_client = None
+
+def init_client():
+    global arxiv_client
+    if arxiv_client is None:
+        arxiv_client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
+    return arxiv_client
+
+def save_story(story, audio_path):
+    """Save story and audio to gallery"""
+    try:
+        # Create gallery directory if it doesn't exist
+        gallery_dir = Path(FILE_DIR_PATH)
+        gallery_dir.mkdir(exist_ok=True)
+        
+        # Generate timestamp for unique filename
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        
+        # Get first line for title
+        first_line = story.split('\n')[0].strip()
+        safe_name = re.sub(r'[^\w\s-]', '', first_line)[:50]
+        
+        # Save story text as markdown
+        story_path = gallery_dir / f"story_{timestamp}_{safe_name}.md"
+        with open(story_path, "w") as f:
+            f.write(f"# {first_line}\n\n{story}")
+        
+        # Copy audio file to gallery
+        new_audio_path = None
+        if audio_path:
+            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) if new_audio_path else None
+    except Exception as e:
+        print(f"Error saving to gallery: {str(e)}")
+        return None, None
 
 def list_all_outputs(generation_history):
     """Load all story generations for community view"""
@@ -57,6 +105,53 @@ def increase_list_size(list_size):
     """Increase the number of visible community generations"""
     return list_size + PAGE_SIZE
 
+def generate_story(prompt, model_choice):
+    """Generate story using specified model"""
+    try:
+        client = init_client()
+        if client is None:
+            return "Error: Story generation service is not available."
+        
+        result = client.predict(
+            prompt=prompt,
+            llm_model_picked=model_choice,
+            stream_outputs=True,
+            api_name="/ask_llm"
+        )
+        return result
+    except Exception as e:
+        return f"Error generating story: {str(e)}"
+
+async def generate_speech(text, voice="en-US-AriaNeural"):
+    """Generate speech from text"""
+    try:
+        communicate = edge_tts.Communicate(text, voice)
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
+            tmp_path = tmp_file.name
+            await communicate.save(tmp_path)
+        return tmp_path
+    except Exception as e:
+        print(f"Error in text2speech: {str(e)}")
+        return None
+
+def process_story_and_audio(prompt, model_choice):
+    """Process story, generate audio, and save to gallery"""
+    try:
+        # Generate story
+        story = generate_story(prompt, model_choice)
+        if isinstance(story, str) and story.startswith("Error"):
+            return story, None
+        
+        # Generate audio
+        audio_path = asyncio.run(generate_speech(story))
+        
+        # Save to gallery
+        story_path, saved_audio_path = save_story(story, audio_path)
+        
+        return story, audio_path
+    except Exception as e:
+        return f"Error: {str(e)}", None
+
 # Add CSS for community generations
 css = '''
 #live_gen:before {
@@ -77,7 +172,7 @@ css = '''
 }
 '''
 
-# Modified Gradio interface
+# Create the Gradio interface
 with gr.Blocks(title="AI Story Generator", css=css) as demo:
     gr.Markdown("""
     # 🎭 AI Story Generator & Narrator
@@ -89,7 +184,6 @@ with gr.Blocks(title="AI Story Generator", css=css) as demo:
     list_size = gr.Number(value=PAGE_SIZE, visible=False)
     
     with gr.Row():
-        # Keep existing left column
         with gr.Column(scale=3):
             with gr.Row():
                 prompt_input = gr.Textbox(
@@ -122,7 +216,7 @@ with gr.Blocks(title="AI Story Generator", css=css) as demo:
                     type="filepath"
                 )
         
-        # Modify right column to include community generations
+        # Sidebar with Story Starters and Community Generations
         with gr.Column(scale=1):
             gr.Markdown("### 📚 Story Starters")
             story_starters = gr.Dataframe(
@@ -131,7 +225,7 @@ with gr.Blocks(title="AI Story Generator", css=css) as demo:
                 interactive=False
             )
             
-            # Add Community Generations section
+            # Community Generations section
             with gr.Column(elem_id="live_gen") as community_list:
                 gr.Markdown("### 🎬 Community Stories")
                 with gr.Column(elem_id="live_gen_items"):
@@ -141,7 +235,7 @@ with gr.Blocks(title="AI Story Generator", css=css) as demo:
                         history_list_latest = history_list[:list_size]
                         
                         for audio_path in history_list_latest:
-                            if not audio_path:
+                            if not audio_path or not os.path.exists(audio_path):
                                 continue
                                 
                             try:
@@ -166,7 +260,7 @@ with gr.Blocks(title="AI Story Generator", css=css) as demo:
                 load_more = gr.Button("Load More Stories")
                 load_more.click(fn=increase_list_size, inputs=list_size, outputs=list_size)
     
-    # Keep existing event handlers
+    # Event handlers
     def update_prompt(evt: gr.SelectData):
         return STORY_STARTERS[evt.index[0]][1]
     
@@ -175,10 +269,10 @@ with gr.Blocks(title="AI Story Generator", css=css) as demo:
     generate_btn.click(
         fn=process_story_and_audio,
         inputs=[prompt_input, model_choice],
-        outputs=[story_output, audio_output, gallery]
+        outputs=[story_output, audio_output]
     )
     
-    # Add auto-refresh for community generations
+    # Auto-refresh for community generations
     demo.load(fn=list_all_outputs, inputs=generation_history, outputs=[generation_history, community_list], every=2)
 
 if __name__ == "__main__":