diff --git a/README.md b/README.md index f169ec8..7f17e67 100644 --- a/README.md +++ b/README.md @@ -7,7 +7,7 @@ A Python script designed to summarize webpages from specified URLs using the Lan [ollama](https://ollama.com/) must be installed and served ```bash -ollama run llama2 +ollama run llama3 ``` ```bash diff --git a/assets/gradio.png b/assets/gradio.png index 159bd18..47f48aa 100644 Binary files a/assets/gradio.png and b/assets/gradio.png differ diff --git a/assets/gradio_old.png b/assets/gradio_old.png new file mode 100644 index 0000000..159bd18 Binary files /dev/null and b/assets/gradio_old.png differ diff --git a/webui.py b/webui.py index e7cc77b..60f8c52 100644 --- a/webui.py +++ b/webui.py @@ -1,21 +1,55 @@ import gradio as gr from summarizer import load_document, setup_summarization_chain +from translator import setup_translator_chain def summarize(url): docs = load_document(url) llm_chain = setup_summarization_chain() result = llm_chain.run(docs) + + return [result, gr.Button("🇹🇷 Translate ", visible=True)] + + +def translate(text): + llm_chain = setup_translator_chain() + result = llm_chain.run(text) return result -iface = gr.Interface( - fn=summarize, - inputs="text", - outputs="text", - title="Summarizer", - description="Enter a URL to get a summary of the document.", -) +with gr.Blocks() as demo: + gr.Markdown( + """# Cobanov Web Summarizer + Easily summarize any web page with a single click.""" + ) -iface.launch() + with gr.Row(): + with gr.Column(): + url = gr.Text(label="URL", placeholder="Enter URL here") + btn_generate = gr.Button("Generate") + + summary = gr.Markdown(label="Summary") + btn_translate = gr.Button(visible=False) + + gr.Examples( + [ + "https://cobanov.dev/haftalik-bulten/hafta-13", + "https://bawolf.substack.com/p/embeddings-are-a-good-starting-point", + ], + inputs=[url], + ) + gr.Markdown( + """ + ``` + Model: llama3-8b + Author: Mert Cobanov + Contact: mertcobanov@gmail.com + Repo: github.com/mertcobanov/easy-web-summarizer + ```""" + ) + btn_generate.click(summarize, inputs=[url], outputs=[summary, btn_translate]) + btn_translate.click(translate, inputs=[summary], outputs=[summary]) + + +demo.launch() diff --git a/webui2.py b/webui2.py deleted file mode 100644 index 60f8c52..0000000 --- a/webui2.py +++ /dev/null @@ -1,55 +0,0 @@ -import gradio as gr - -from summarizer import load_document, setup_summarization_chain -from translator import setup_translator_chain - - -def summarize(url): - docs = load_document(url) - llm_chain = setup_summarization_chain() - result = llm_chain.run(docs) - - return [result, gr.Button("🇹🇷 Translate ", visible=True)] - - -def translate(text): - llm_chain = setup_translator_chain() - result = llm_chain.run(text) - return result - - -with gr.Blocks() as demo: - gr.Markdown( - """# Cobanov Web Summarizer - Easily summarize any web page with a single click.""" - ) - - with gr.Row(): - with gr.Column(): - url = gr.Text(label="URL", placeholder="Enter URL here") - btn_generate = gr.Button("Generate") - - summary = gr.Markdown(label="Summary") - btn_translate = gr.Button(visible=False) - - gr.Examples( - [ - "https://cobanov.dev/haftalik-bulten/hafta-13", - "https://bawolf.substack.com/p/embeddings-are-a-good-starting-point", - ], - inputs=[url], - ) - gr.Markdown( - """ - ``` - Model: llama3-8b - Author: Mert Cobanov - Contact: mertcobanov@gmail.com - Repo: github.com/mertcobanov/easy-web-summarizer - ```""" - ) - btn_generate.click(summarize, inputs=[url], outputs=[summary, btn_translate]) - btn_translate.click(translate, inputs=[summary], outputs=[summary]) - - -demo.launch() diff --git a/webui_old.py b/webui_old.py new file mode 100644 index 0000000..e7cc77b --- /dev/null +++ b/webui_old.py @@ -0,0 +1,21 @@ +import gradio as gr + +from summarizer import load_document, setup_summarization_chain + + +def summarize(url): + docs = load_document(url) + llm_chain = setup_summarization_chain() + result = llm_chain.run(docs) + return result + + +iface = gr.Interface( + fn=summarize, + inputs="text", + outputs="text", + title="Summarizer", + description="Enter a URL to get a summary of the document.", +) + +iface.launch()