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)