testStreamlit / app.py
farukkr's picture
Update app.py
c9a5e55 verified
raw
history blame contribute delete
863 Bytes
import streamlit as st
import requests
API_URL = "https://jj78yliqd7.execute-api.eu-central-1.amazonaws.com/dev/chatbot"
def get_answer(prompt):
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."
# Streamlit App
st.sidebar.title("Wissensguru - IDTA")
# Upload-Bereich
uploaded_file = st.file_uploader("Upload PDF file")
if uploaded_file is not None:
upload_file(uploaded_file)
# Chatbot-Bereich
st.subheader("Chatbot")
user_question = st.text_input("Ask me a question")
if st.button("Ask"):
if user_question:
answer = get_answer(user_question)
st.write("Answer:", answer)