print response from model

app.py

@@ -57,6 +57,26 @@ def load_model():
     except Exception as e:
         print(f"Error loading model: {e}")
         return None
+    
+# 모델 응답 처리
+def get_response_from_model(prompt):
+    try:
+        if "knowledge_base" not in st.session_state:
+            return "No PDF has been uploaded yet."
+        if "qa_chain" not in st.session_state:
+            return "QA chain is not initialized."
+
+        docs = st.session_state.knowledge_base.similarity_search(prompt)
+        print("docs:", docs)
+        response = st.session_state.qa_chain.run(input_documents=docs, question=prompt)
+        print("response2:", response)
+
+        if "Helpful Answer:" in response:
+            response = response.split("Helpful Answer:")[1].strip()
+
+        return response
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 
 # 페이지 UI
@@ -126,24 +146,6 @@ def main():
                 if response:
                     st.write(f"**Assistant**: {response}")
 
-# 모델 응답 처리
-def get_response_from_model(prompt):
-    try:
-        if "knowledge_base" not in st.session_state:
-            return "No PDF has been uploaded yet."
-        if "qa_chain" not in st.session_state:
-            return "QA chain is not initialized."
-
-        docs = st.session_state.knowledge_base.similarity_search(prompt)
-        response = st.session_state.qa_chain.run(input_documents=docs, question=prompt)
-
-        if "Helpful Answer:" in response:
-            response = response.split("Helpful Answer:")[1].strip()
-
-        return response
-    except Exception as e:
-        return f"Error: {str(e)}"
-
 
 # 앱 실행
 if __name__ == "__main__":