Update app.py

app.py

@@ -221,9 +221,13 @@ with gr.Blocks() as agent_ui:
 
         code, preview, words, keywords = generate_blogger_html(app_query, custom_link, mod_features, t_color, t_link, c_size)
         
-        # Unhide the file download component so the user can download the txt file
-        return code, preview, words, keywords, gr.update(visible=False), gr.update(visible=True), state, gr.update(value="latest_draft.txt", visible=True)
-
+        # --- THE FIX: Only attempt to serve the file if it exists and there are no errors ---
+        if os.path.exists("latest_draft.txt") and not code.startswith("Error"):
+            file_update = gr.update(value="latest_draft.txt", visible=True)
+        else:
+            file_update = gr.update(value=None, visible=False)
+            
+        return code, preview, words, keywords, gr.update(visible=False), gr.update(visible=True), state, file_update
     # --- BINDS ---
     login_btn.click(fn=verify_login, inputs=[pwd_input, session_state], outputs=[session_state, login_screen, main_app, login_err])