find int error

app.py

@@ -74,15 +74,18 @@ def get_response_from_model(prompt):
 
         docs = st.session_state.knowledge_base.similarity_search(prompt)
         print("docs:", docs) # 이까진 됐는데
+        print("prompt:", prompt) # 이까진 됐는데
         # Chain의 invoke() 메소드 사용 (input_documents로 전달)
         response = st.session_state.qa_chain.invoke({
             "input_documents": docs,
             "question": prompt
         })
-
-        if "Helpful Answer:" in response:
-            response = response.split("Helpful Answer:")[1].strip()
-
+        try:
+            if "Helpful Answer:" in response:
+                response = response.split("Helpful Answer:")[1].strip()
+        except ValueError as e:
+            print(f"ValueError occurred: {e}")
+            return f"Error: Invalid response format - {e}"
         return response
     except Exception as e:
         return f"Error: {str(e)}"
@@ -129,7 +132,6 @@ def main():
                             st.error("Failed to create knowledge base.")
                             return
 
-                        # QA 체인 설정
                         try:
                             pipe = load_model()
                         except Exception as e:
@@ -147,15 +149,11 @@ def main():
     if "knowledge_base" in st.session_state and "qa_chain" in st.session_state:
         with st.expander("Ask Questions", expanded=True):
             prompt = st.text_input("Chat here!")
-
             if prompt:
-                print("prompt:", prompt)
                 response = get_response_from_model(prompt)
-                print("Response:", response)
                 if response:
                     st.write(f"**Assistant**: {response}")
 
-
 # 앱 실행
 if __name__ == "__main__":
     main()