| import gradio as gr |
| import requests |
| import json |
|
|
| API_URL = "https://jj78yliqd7.execute-api.eu-central-1.amazonaws.com/dev/chatbot" |
|
|
| def get_answer(prompt, message): |
| params = {"prompt": prompt} |
| response = requests.get(API_URL, params=params) |
| if response.status_code == 200: |
| response_body = response.json().get('body') |
| answer = response_body.get('answer') |
| return answer |
| else: |
| return "Sorry, something went wrong. Please try again." |
|
|
| def upload_file(file): |
| pdf_file_path = file.name |
| with open(pdf_file_path, 'rb') as pdf_file: |
| pdf_binary_data = pdf_file.read() |
| |
| API_ENDPOINT = "https://jj78yliqd7.execute-api.eu-central-1.amazonaws.com/dev/chatbot" |
|
|
| headers = { |
| 'Content-Type': 'application/pdf' |
| } |
| |
| pdf_file_path = pdf_file_path.replace("\\", "/").split("/")[-1] |
| |
| params = {'filename': pdf_file_path} |
|
|
| response = requests.post(API_ENDPOINT, data=pdf_binary_data, headers=headers, params=params) |
| if response.status_code == 200: |
| print("File successfully uploaded") |
| else: |
| print(f"Failed to upload file. Status code: {response.status_code}, Response: {response.text}") |
|
|
|
|
|
|
| chat = gr.ChatInterface( |
| get_answer, |
| chatbot=gr.Chatbot(height=700), |
| textbox=gr.Textbox(placeholder="Ask me a question", container=False, scale=7), |
| title="Wissensguru - IDTA", |
| theme="soft", |
| examples=[ |
| "When was the first version of this document produced?" |
| ], |
| cache_examples=False, |
| retry_btn=None, |
| undo_btn="Delete Previous", |
| clear_btn="Clear", |
| ) |
|
|
| with gr.Blocks() as bot: |
| gr.Markdown("Upload pdf files") |
| with gr.Tab("Pdf Upload"): |
| file_output = gr.File(file_types=["file"]) |
| file_output.upload(upload_file, file_output) |
| with gr.Tab("Chatbot"): |
| chat.render() |
| |
| bot.launch() |
|
|
