Create app.py

app.py

@@ -0,0 +1,50 @@
+import streamlit as st
+
+
+icons = {"assistant": "robot.png", "user": "man-kddi.png"}
+
+def streamer(text):
+    for i in text:
+        yield i
+        time.sleep(0.005)
+
+# Streamlit app initialization
+st.title("RAG AGENT")
+st.markdown("RAG Agent with PDF and Web Search")
+
+if 'messages' not in st.session_state:
+    st.session_state.messages = [{'role': 'assistant', "content": 'Hello! Upload a PDF/Youtube Video link and ask me anything about the content.'}]
+
+for message in st.session_state.messages:
+    with st.chat_message(message['role'], avatar=icons[message['role']]):
+        st.write(message['content'])
+
+with st.sidebar:
+    st.title("Menu:")
+    uploaded_file = st.file_uploader("Upload your PDF Files and Click on the Submit & Process Button")
+    video_url = st.text_input("Enter Youtube Video Link: ")
+    if st.button("Submit & Process"):
+        with st.spinner("Processing..."):
+            print("Processing files")
+                
+            if uploaded_file:
+                print("File uploaded")
+        
+            if video_url:
+                print("Link uploaded:"+video_url)
+
+            st.success("Done")
+
+user_prompt = st.chat_input("Ask me anything about the content of the PDF or Web Link:")
+
+if user_prompt and (uploaded_file or video_url):
+    st.session_state.messages.append({'role': 'user', "content": user_prompt})
+    with st.chat_message("user", avatar="man-kddi.png"):
+        st.write(user_prompt)
+
+    # Trigger assistant's response retrieval and update UI
+    with st.spinner("Thinking..."):
+        response = "I have an answer coming soon..."
+    with st.chat_message("user", avatar="robot.png"):
+        st.write_stream(streamer(response))
+    st.session_state.messages.append({'role': 'assistant', "content": response})