Update chatbot.py

chatbot.py

@@ -211,7 +211,6 @@ def cleanup_response(response):
         response = response[answer_start + len("Answer:"):].strip()
     return response
 
-# Gradio interface for the chatbot
 def chatbot(audio, input_type, text):
     if input_type == "Voice":
         transcription = query_whisper(audio.name)
@@ -221,15 +220,20 @@ def chatbot(audio, input_type, text):
     else:
         query = text
 
+    # Extract details from the prompt
     details = extract_details_from_prompt(query)
+    
+    # Get aggregated patient history based on the extracted details
     patient_history = get_aggregated_patient_history(patient_data, details)
 
+    # Create the payload with the patient history and the user's query
     payload = {
         "inputs": f"role: ophthalmologist assistant patient history: {patient_history} question: {query}"
     }
 
     logging.debug(f"Raw input to the LLM: {payload['inputs']}")
 
+    # Query the Hugging Face model with the payload
     response = query_huggingface(payload)
     if isinstance(response, list):
         raw_response = response[0].get("generated_text", "Sorry, I couldn't generate a response.")
@@ -238,8 +242,7 @@ def chatbot(audio, input_type, text):
 
     logging.debug(f"Raw output from the LLM: {raw_response}")
 
-    clean_response = cleanup_response(raw_response)
-    return clean_response, None
+    return raw_response, None
 
 # Gradio interface for generating voice response
 def generate_voice_response(tts_model, text_response):