Update app.py

app.py

@@ -70,7 +70,27 @@ def generate_story(file_paths, progress=gr.Progress(track_tqdm=True)):
 
     return story
 
-with gr.Blocks() as demo:
+const footerMD2 = """
+@misc{oliveira2025storyreasoningdatasetusingchainofthought,
+      title={StoryReasoning Dataset: Using Chain-of-Thought for Scene Understanding and Grounded Story Generation}, 
+      author={Daniel A. P. Oliveira and David Martins de Matos},
+      year={2025},
+      eprint={2505.10292},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV},
+      url={https://arxiv.org/abs/2505.10292}, 
+}
+"""
+const footerMD = """
+### Key Features
+* Cross-Frame Consistency: Maintains consistent character and object identity across multiple frames through visual similarity and face recognition techniques
+* Structured Reasoning: Employs chain-of-thought reasoning to analyze scenes with explicit modeling of characters, objects, settings, and narrative structure
+* Grounded Storytelling: Uses specialized XML tags to link narrative elements directly to visual entities
+* Reduced Hallucinations: Achieves 12.3% fewer hallucinations compared to the non-fine-tuned base model
+
+Model trained by daniel3303, [repository here.](https://huggingface.co/daniel3303/QwenStoryteller)
+"""
+with gr.Blocks(fill_height=True) as demo:
     gr.Markdown("# Qwen Storyteller \n## Upload up to 6 images to generate a creative story.")
     
     with gr.Row():
@@ -79,8 +99,12 @@ with gr.Blocks() as demo:
             output_file = gr.File(label="Uploaded Files")
 
         with gr.Column():
-            outputs = gr.Textbox(label="Generated Story", lines=10)
+            outputs = gr.Markdown(label="Generated Story", show_copy_button=True)
 
+    with gr.Row():
+        gr.Markdown(footerMD)
+        gr.Markdown(footerMD2)
+        
     upload_button.upload(lambda files: [f.name for f in files], upload_button, output_file)
 
     gen_button = gr.Button("Generate", variant="secondary")