Spaces:
Running
Running
| import gradio as gr | |
| import worker | |
| import requests | |
| from pathlib import Path | |
| import torchvision | |
| torchvision.disable_beta_transforms_warning() | |
| # Get data from url | |
| url = 'https://camels.readthedocs.io/_/downloads/en/latest/pdf/' | |
| r = requests.get(url, stream=True) | |
| document_path = Path('metadata.pdf') | |
| document_path.write_bytes(r.content) | |
| # document_path="2022GS.pdf" | |
| worker.process_document(document_path) | |
| def handle_prompt(message, history): | |
| bot_response = worker.process_prompt(message, history) | |
| return bot_response | |
| greetingsmessage = "Hi, I'm the CAMELS DocBot, I'm here to assist you with any question related to the CAMELS simulations documentation" | |
| example_questions = [ | |
| "How can i read a halo file?", | |
| "Which simulation suites are included in CAMELS?", | |
| "Which are the largest volumes in CAMELS simulations?", | |
| "How can I get the power spectrum of a simulation?" | |
| ] | |
| # chatbot = gr.Chatbot(value=[{"role": "assistant", "content": greetingsmessage}]) | |
| # chatbot = gr.Chatbot(value=[[None, greetingsmessage]]) | |
| # chatbot = gr.Chatbot(value=gr.ChatMessage(role="assistant",content="How can I help you?")) | |
| # chatbot = gr.Chatbot(placeholder=greetingsmessage) | |
| demo = gr.ChatInterface(handle_prompt, type="messages", title="CAMELS DocBot",examples=example_questions, theme=gr.themes.Soft(), description=greetingsmessage)#, chatbot=chatbot) | |
| demo.launch() |