Update app.py

app.py

@@ -32,6 +32,7 @@ from openai import ChatCompletion
 from PyPDF2 import PdfReader
 from templates import bot_template, css, user_template
 from xml.etree import ElementTree as ET
+import streamlit.components.v1 as components  # Import Streamlit Components for HTML5
 
 def add_Med_Licensing_Exam_Dataset():
     import streamlit as st
@@ -178,7 +179,6 @@ def addDocumentHTML5(result):
     </html>
     '''
 
-    import streamlit.components.v1 as components  # Import Streamlit
     components.html(documentHTML5, width=1280, height=1024)
     return result
 
@@ -186,7 +186,6 @@ def addDocumentHTML5(result):
 # 3. Stream Llama Response
 # @st.cache_resource
 def StreamLLMChatResponse(prompt):
-
     try:
         endpoint_url = API_URL
         hf_token = API_KEY
@@ -221,7 +220,8 @@ def StreamLLMChatResponse(prompt):
                     
             except:
                 st.write('Stream llm issue')
-        add_documentHTML5(result)
+        #add_documentHTML5(result)
+        return result
     except:
         st.write('Llama model is asleep. Starting up now on A10 - please give 5 minutes then retry as KEDA scales up from zero to activate running container(s).')