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"""
-
""",
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)