tcsenpai/youlama
1
0
Fork 0
mirror of https://github.com/tcsenpai/youlama.git synced 2025-06-06 19:25:39 +00:00
Code Issues 1 Packages Projects Releases Wiki Activity

better layout

Browse Source
This commit is contained in:
tcsenpai 2024-12-25 18:41:02 +01:00
parent f64b409805
commit 0b5ffc05a7
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/main.py
View File

@ -177,7 +177,7 @@ def show_error(message):
def show_info(message): def show_info(message):
update_header("<EFBFBD><EFBFBD><EFBFBD> " + message) update_header("<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> " + message)
def update_header(message): def update_header(message):
@ -410,10 +410,12 @@ def main():
if rephrase_button: if rephrase_button:
# Only rephrase the transcript # Only rephrase the transcript
show_warning("Starting rephrasing, this might take a while...") show_warning("Starting grammar and punctuation fixes...")
with st.spinner("Rephrasing transcript..."): with st.spinner("Fixing transcript..."):
ollama_client = OllamaClient(ollama_url, selected_model) ollama_client = OllamaClient(ollama_url, selected_model)
prompt = f"Rephrase the following transcript to make it more readable and well-formatted, keeping the main content intact:\n\n{transcript}" prompt = f"""Fix the grammar and punctuation of the following transcript, maintaining the exact same content and meaning.
Only correct grammatical errors, add proper punctuation, and fix sentence structure where needed.
Do not rephrase or change the content:\n\n{transcript}"""
rephrased = ollama_client.generate(prompt) rephrased = ollama_client.generate(prompt)
video_info = get_video_info(video_id) video_info = get_video_info(video_id)
@ -426,7 +428,7 @@ def main():
with info_col2: with info_col2:
st.write(f"**Channel:** {video_info['channel']}") st.write(f"**Channel:** {video_info['channel']}")
st.subheader("📝 Rephrased Transcript") st.subheader("📝 Fixed Transcript")
st.markdown(rephrased) st.markdown(rephrased)
elif summarize_button: elif summarize_button: