wrap code lines

app.py

@@ -1070,7 +1070,7 @@ def main():
                     st.caption(f"Prompt tokens: {prompt_tokens:,}")
                 # Make prompt scrollable using container
                 with st.container(height=500):
-                    st.code(results["prompt"], language="text")
+                    st.code(results["prompt"], language="markdown", wrap_lines=True)
                 st.divider()
             
             # Raw output section
@@ -1089,7 +1089,7 @@ def main():
                     token_counter = TokenCounter(model=tokenizer_name)
                     output_tokens = token_counter(content)
                     st.caption(f"Output tokens: {output_tokens:,}")
-                st.code(content, language="yaml")
+                st.code(content, language="yaml", wrap_lines=True)
                 st.divider()
             
             # Step timing information