Update app.py

app.py

@@ -33,6 +33,7 @@ if st.button("Send"):
         A headache can have many causes, ranging from stress, dehydration, or fatigue to more severe conditions like migraines, infections, or neurological problems. Common remedies include over-the-counter pain relievers, hydration, and rest. 
         If headaches are persistent or severe, it may indicate an underlying condition such as tension headaches, cluster headaches, or even infections like sinusitis. If the headache is accompanied by other symptoms such as nausea, vision changes, or confusion, it is recommended to seek medical attention.
         """
+        
         inputs = medical_tokenizer.encode_plus(user_input, context, add_special_tokens=True, return_tensors="pt")
         input_ids = inputs["input_ids"].tolist()[0]
 
@@ -44,16 +45,17 @@ if st.button("Send"):
 
         answer_start = torch.argmax(answer_start_scores)
         answer_end = torch.argmax(answer_end_scores) + 1
+        
         medical_answer = medical_tokenizer.convert_tokens_to_string(medical_tokenizer.convert_ids_to_tokens(input_ids[answer_start:answer_end]))
 
-        # If no medical answer is found, provide a fallback response
-        if medical_answer.strip() == "":
+        # Check if a valid medical answer was found
+        if not medical_answer.strip():  # If the answer is empty or only whitespace
             medical_answer = "I'm not sure about that. You may want to consult a medical professional."
-
+        
         # Append medical response to the conversation history
         st.session_state.conversation_history.append(f"Bot (Medical): {medical_answer}")
 
-        # DialoGPT for conversational response
+        # Generate conversational response with DialoGPT
         conversation_input_ids = conversation_tokenizer.encode(user_input + conversation_tokenizer.eos_token, return_tensors='pt')
         conversation_bot_input_ids = torch.cat([conversation_tokenizer.encode(convo + conversation_tokenizer.eos_token, return_tensors='pt') for convo in st.session_state.conversation_history], dim=-1)
 
@@ -64,8 +66,11 @@ if st.button("Send"):
         # Append conversational response to conversation history
         st.session_state.conversation_history.append(f"Bot (Conversational): {conversation_response}")
 
-        # Display conversation history
+        # Display the entire conversation history
         for message in st.session_state.conversation_history:
             st.write(message)
     else:
         st.write("Please enter a medical question.")
+
+
+