Update src/streamlit_app.py

src/streamlit_app.py

@@ -107,9 +107,12 @@ parser = StrOutputParser()
 
 def create_rag_chain(retriever, model, prompt):
     def fetch_context(user_input):
+        if isinstance(user_input, dict):
+            user_input = user_input.get("question", "")
         docs = retriever.get_relevant_documents(user_input)
-        context = "\n\n".join([doc.page_content for doc in docs])
-        return {"context": context, "question": user_input}
+        context_text = "\n\n".join(doc.page_content for doc in docs)
+        return {"context": context_text, "question": user_input}
+
 
     chain = fetch_context | prompt | model | parser
     return chain