Update app.py

app.py

@@ -13,17 +13,23 @@ import json
 
 warnings.filterwarnings('ignore')
 
-# Initialize client outside of reload block
-if gr.NO_RELOAD:
-    ARXIV_CLIENT = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
+# 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 generate_story(prompt, model_choice):
     """Generate story using specified model"""
     try:
-        if ARXIV_CLIENT is None:
+        client = init_client()
+        if client is None:
             return "Error: Story generation service is not available."
         
-        result = ARXIV_CLIENT.predict(
+        result = client.predict(
             prompt=prompt,
             llm_model_picked=model_choice,
             stream_outputs=True,
@@ -50,7 +56,7 @@ def process_story_and_audio(prompt, model_choice):
     try:
         # Generate story
         story = generate_story(prompt, model_choice)
-        if story.startswith("Error"):
+        if isinstance(story, str) and story.startswith("Error"):
             return story, None
 
         # Generate audio
@@ -60,57 +66,49 @@ def process_story_and_audio(prompt, model_choice):
     except Exception as e:
         return f"Error: {str(e)}", None
 
-# Define the Gradio interface
-def create_app():
-    with gr.Blocks(title="AI Story Generator") as demo:
-        gr.Markdown("""
-        # 🎭 AI Story Generator & Narrator
-        Generate creative stories and listen to them!
-        """)
-        
-        with gr.Row():
-            with gr.Column():
-                prompt_input = gr.Textbox(
-                    label="Story Concept",
-                    placeholder="Enter your story idea...",
-                    lines=3
-                )
-                model_choice = gr.Dropdown(
-                    label="Model",
-                    choices=[
-                        "mistralai/Mixtral-8x7B-Instruct-v0.1",
-                        "mistralai/Mistral-7B-Instruct-v0.2"
-                    ],
-                    value="mistralai/Mixtral-8x7B-Instruct-v0.1"
-                )
-                generate_btn = gr.Button("Generate Story")
-        
-        with gr.Row():
-            story_output = gr.Textbox(
-                label="Generated Story",
-                lines=10,
-                interactive=False
+# Create the Gradio interface
+with gr.Blocks(title="AI Story Generator") as demo:
+    gr.Markdown("""
+    # 🎭 AI Story Generator & Narrator
+    Generate creative stories and listen to them!
+    """)
+    
+    with gr.Row():
+        with gr.Column():
+            prompt_input = gr.Textbox(
+                label="Story Concept",
+                placeholder="Enter your story idea...",
+                lines=3
             )
-        
-        with gr.Row():
-            audio_output = gr.Audio(
-                label="Story Narration",
-                type="filepath"
+            model_choice = gr.Dropdown(
+                label="Model",
+                choices=[
+                    "mistralai/Mixtral-8x7B-Instruct-v0.1",
+                    "mistralai/Mistral-7B-Instruct-v0.2"
+                ],
+                value="mistralai/Mixtral-8x7B-Instruct-v0.1"
             )
-        
-        generate_btn.click(
-            fn=process_story_and_audio,
-            inputs=[prompt_input, model_choice],
-            outputs=[story_output, audio_output]
+            generate_btn = gr.Button("Generate Story")
+    
+    with gr.Row():
+        story_output = gr.Textbox(
+            label="Generated Story",
+            lines=10,
+            interactive=False
+        )
+    
+    with gr.Row():
+        audio_output = gr.Audio(
+            label="Story Narration",
+            type="filepath"
         )
     
-    return demo
+    generate_btn.click(
+        fn=process_story_and_audio,
+        inputs=[prompt_input, model_choice],
+        outputs=[story_output, audio_output]
+    )
 
-# Launch the app
+# Launch the app using the current pattern
 if __name__ == "__main__":
-    demo = create_app()
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True
-    )
+    demo.launch()