tcsenpai/youlama
1
0
Fork 0
mirror of https://github.com/tcsenpai/youlama.git synced 2025-06-07 03:35:41 +00:00
Code Issues 1 Packages Projects Releases Wiki Activity

better title styling

Browse Source
This commit is contained in:
tcsenpai 2024-12-25 21:18:32 +01:00
parent 5fc59a9610
commit e27faa9f34
1 changed files with 38 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
src/main.py
View File

@ -37,10 +37,10 @@ def main():
# Load CSS # Load CSS
load_css() load_css()
st.write("#### YouTube Video Companion") # st.write("###### YouTube Video Companion")
# Ollama Settings section # Ollama Settings section
#st.subheader("🎯 Ollama Settings") # st.subheader("🎯 Ollama Settings")
default_ollama_url = os.getenv("OLLAMA_URL") default_ollama_url = os.getenv("OLLAMA_URL")
ollama_url = st.text_input( ollama_url = st.text_input(
@ -117,35 +117,45 @@ def main():
update_header("🚫 " + message) update_header("🚫 " + message)
def show_info(message): def show_info(message):
update_header("> " + message) update_header("💡 " + message)
def update_header(message): def update_header(message):
with header: with header:
st.markdown( st.markdown(
f""" f"""
<div class='fixed-header'> <div class='fixed-header'>
{message} <div class='header-message'>{message}</div>
<div class='header-title'>YouLama - A YouTube Video Companion</div>
<style>
div.fixed-header {{
position: fixed;
top: 2.875rem;
left: 0;
right: 0;
z-index: 999;
padding: 10px;
margin: 0 1rem;
border-radius: 0.5rem;
border: 1px solid rgba(128, 128, 128, 0.2);
height: 45px !important;
background-color: rgba(40, 40, 40, 0.95);
backdrop-filter: blur(5px);
box-shadow: 0 4px 6px rgb(0, 0, 0, 0.1);
transition: all 0.3s ease;
display: flex;
align-items: center;
justify-content: space-between;
}}
.header-message {{
flex: 1;
}}
.header-title {{
color: rgba(250, 250, 250, 0.8);
font-size: 0.9em;
margin-left: 20px;
}}
</style>
</div> </div>
<style>
div.fixed-header {{
position: fixed;
top: 2.875rem;
left: 0;
right: 0;
z-index: 999;
padding: 10px;
margin: 0 1rem;
border-radius: 0.5rem;
border: 1px solid rgba(128, 128, 128, 0.2);
height: 45px !important;
background-color: rgba(40, 40, 40, 0.95);
backdrop-filter: blur(5px);
box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
transition: all 0.3s ease;
display: flex;
align-items: center;
}}
</style>
""", """,
unsafe_allow_html=True, 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:" 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) summary = ollama_client.generate(prompt)
print(summary) print(summary)
show_info("Summary generated successfully!") show_info("Summary generated successfully (scroll down to see the summary)!")
with st.spinner("Fetching video info..."): with st.spinner("Fetching video info..."):
video_info = get_video_info(video_id) 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. Only correct grammatical errors, add proper punctuation, and fix sentence structure where needed.
Do not rephrase or change the content:\n\n{transcript}""" Do not rephrase or change the content:\n\n{transcript}"""
enhanced = ollama_client.generate(prompt) 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..."): with st.spinner("Fetching video info..."):
video_info = get_video_info(video_id) video_info = get_video_info(video_id)