Update app.py

app.py

@@ -143,9 +143,9 @@ with col2:
     if search_input:
         with st.spinner("Generating response..."):
             try:
-                if model_selection == "Disease Management":
+                if model_selection == "Disease Analysis":
                     pipe = create_pipeline("harishussain12/Disease_Managment")
-                    
+
                     context = ""
                     if uploaded_file is not None:
                         file_content = read_pdf(uploaded_file)
@@ -155,6 +155,8 @@ with col2:
                             context = file_content
 
                     query_input = search_input + (f"\n\nContext:\n{context}" if context else "")
+                    st.write("Debug: Query Input:", query_input)  # Debugging
+
                     response = pipe(query_input, max_length=200, num_return_sequences=1)
                     st.markdown(f"### Response:\n{response[0]['generated_text']}")
 
@@ -174,4 +176,4 @@ with col2:
                         st.markdown(f"### Answer:\n{result['answer']}")
 
             except Exception as e:
-                st.error(f"Error generating response: {str(e)}")
+                st.error(f"Error generating response: {str(e)}")