import httpx import gradio as gr def submit(inputs): if not inputs: return payload = {"content": inputs, "author": "anna nymous"} try: httpx.post("http://localhost:8080/submit/", json=payload) except httpx.ConnectError: pass def check_status(): try: response = httpx.get("http://localhost:8080/check_job_status/") result = response.json() except httpx.ConnectError: result = "Server could not be reached" return result def check_tags(): url = "http://localhost:8080/tag_counts/" try: tag_counts = { key.strip("#"): val for key, val in httpx.get(url).json().items() if key.strip("#") } sorted_tags = sorted(list(tag_counts.items()), key=lambda tup: -tup[1]) result = "Most common tags: " + ", ".join( f"{t} ({c})" for t, c in sorted_tags[:5] ) except httpx.ConnectError: result = "" return result def get_results(inputs: list[str]): if not inputs: response = httpx.get("http://localhost:8080/recent/") else: tags = inputs.replace(",", " ").split() tags = [tag.lstrip("#") for tag in tags] response = httpx.get("http://localhost:8080/recent/" + ",".join(tags)) entries = response.json() texts: list[str] = [] for i, entry in enumerate(entries, start=1): texts.append( f"## {i}. author: {entry['author']}\n\n" f"Date: _{entry['date']}_\n\n" f"Source: _{entry['source_snippet']}_\n\n" f"**Summary**: {entry['summary']}\n\n" f"tags: _{' '.join(entry['tags'])}_" ) return "\n\n---\n\n".join(texts) INPUT_DESCRIPTION = """# Gistillery The no-fuss knowledge dump. Paste information into the field below and submit, _Gistillery_ will create a short summary and hashtags for it. Then show results for most recent entries, optionally filtered by hashtag. Input currently supports: - plain text - a URL to a webpage - a URL to a youtube video (the video will be transcribed) - a URL to an image (url ending in .png, .jpg, etc.; the image description will be used) """ def get_demo(): with gr.Blocks() as demo: # submit new input gr.Markdown(INPUT_DESCRIPTION) inputs = gr.Textbox(lines=3, label="Input") btn_submit = gr.Button("Submit") # clear text box btn_submit.click(lambda x: gr.update(value=''), [], [inputs]) inputs.submit(lambda x: gr.update(value=''), [], [inputs]) # check job status gr.Markdown( "Processing can take a couple of minutes, the info box below gives an " "update on how many jobs are in the queue." ) gr.HTML(value=check_status, label="Status", every=3) gr.HTML(value=check_tags, label="Status", every=9) # check box of tags to filter on msg = "Enter hashtags to filter on (comma separated if multiple)" tags = gr.Textbox(label=msg) # display output btn_output = gr.Button("Show results") output = gr.Markdown() btn_submit.click(submit, inputs=inputs) btn_output.click(get_results, inputs=[tags], outputs=[output]) return demo if __name__ == "__main__": demo = get_demo() demo.queue() demo.launch()