diff --git a/src/main.py b/src/main.py index 4ec5a60..933dc56 100644 --- a/src/main.py +++ b/src/main.py @@ -37,10 +37,10 @@ def main(): # Load CSS load_css() - st.write("#### YouTube Video Companion") + # st.write("###### YouTube Video Companion") # Ollama Settings section - #st.subheader("🎯 Ollama Settings") + # st.subheader("🎯 Ollama Settings") default_ollama_url = os.getenv("OLLAMA_URL") ollama_url = st.text_input( @@ -117,35 +117,45 @@ def main(): update_header("🚫 " + message) def show_info(message): - update_header("> " + message) + update_header("💡 " + message) def update_header(message): with header: st.markdown( f"""
- {message} +
{message}
+
YouLama - A YouTube Video Companion
+
- """, unsafe_allow_html=True, ) @@ -242,7 +252,7 @@ def main(): prompt = f"Summarize the following YouTube video transcript in a concise yet detailed manner:\n\n```{transcript}```\n\nSummary with introduction and conclusion formatted in markdown:" summary = ollama_client.generate(prompt) print(summary) - show_info("Summary generated successfully!") + show_info("Summary generated successfully (scroll down to see the summary)!") with st.spinner("Fetching video info..."): video_info = get_video_info(video_id) @@ -317,7 +327,9 @@ def main(): Only correct grammatical errors, add proper punctuation, and fix sentence structure where needed. Do not rephrase or change the content:\n\n{transcript}""" enhanced = ollama_client.generate(prompt) - show_info("Transcript enhanced successfully!") + show_info( + "Transcript enhanced successfully (scroll down to see the enhanced transcript)!" + ) with st.spinner("Fetching video info..."): video_info = get_video_info(video_id)