from Ascle import Ascle

# create Ascle 
med = Ascle()

# Text Simplification
main_record = """
              The patient presents with symptoms of acute bronchitis,
              including cough, chest congestion, and mild fever.
              Auscultation reveals coarse breath sounds and occasional 
              wheezing. Based on the clinical examination, a diagnosis
              of acute bronchitis is made, and the patient is prescribed 
              a short course of bronchodilators and advised to rest and
              stay hydrated.
              """

# choose the model
layman_model = "ireneli1024/bart-large-elife-finetuned"

med.update_and_delete_main_record(main_record)

# call the text simplification function and print the output
print(med.get_layman_text(layman_model, min_length=20, max_length=70))

>> """
   The patient presents with symptoms of acute bronchitis including
   cough, chest congestion and mild fever. Auscultation reveals coarse 
   breath sounds and occasional wheezing. Based on these symptoms and 
   the patient's history of previous infections with the same condition, 
   the doctor decides that the patient is likely to have a cold or bronch.
   """

main_record = """
              Myeloid derived suppressor cells (MDSC) are immature myeloid 
              cells with immunosuppressive activity. They accumulate in 
              tumor-bearing mice and humans with different types of cancer, 
              including hepatocellular carcinoma (HCC).
              """
              
med.update_and_delete_main_record(main_record)

# call the machine translation function and print the output
print(med.get_translation_mt5("French"))

>> """
   Les cellules suppressives dérivées de myéloïdes (MDSC) sont des
   cellules myéloïdes immatures ayant une activité immunosuppressive, 
   accumulées chez des souris et des humains ayant différents types de 
   cancer, y compris le carcinome hépatocellulaire (HCC).
   """

main_record =  """    
               summarize this text: 
               Neurons (also called neurones or nerve cells) are the fundamental units of the brain and nervous system,  
               the cells responsible for receiving sensory input from the external world, for sending motor commands to  
               our muscles, and for transforming and relaying the electrical signals at every step in between. More than  
               that, their interactions define who we are as people. Having said that, our roughly 100 billion neurons do 
               interact closely with other cell types, broadly classified as glia (these may actually outnumber neurons,  
               although it’s not really known)
               """
              
med.update_and_delete_main_record(main_record)

# call the GPT function and print the output
print(med.call_GPT(api_key="xxxx"))

>> """
   Neurons, or nerve cells, are essential units of the brain and nervous 
   system responsible for receiving sensory input, sending motor commands 
   to muscles and processing electrical signals. They also play a significant 
   role in defining human personality. Despite being approximately 100 billion
   in number, neurons often interact with other cell types known as glia, which
   may outnumber neurons.
   """

              
med.update_and_delete_main_record(main_record)

# call the Claude function and print the output
print(med.call_Claude(api_key="xxxx"))

>> """
   Here is a summary of the text:

   - Neurons are the basic cells of the brain and nervous system.
   - They receive sensory input, send motor signals to muscles, and relay electrical signals in between.
   - Neuron interactions shape our identity and personality.
   - There are about 100 billion neurons in the human brain.
   - Neurons interact with glial cells, which may outnumber neurons.
   """

              
med.update_and_delete_main_record(main_record)

# call the Gemini function and print the output
print(med.call_Gemini(api_key="xxxx"))

>> """   
   Neurons, the building blocks of the brain and nervous system,
   are responsible for receiving sensory information, sending motor 
   signals, and transmitting electrical signals throughout the 
   body. Their intricate interactions shape our identities. While 
   there are approximately 100 billion neurons, they work closely 
   with glial cells, which may even outnumber neurons but whose 
   exact quantity is unknown.
   """

              
med.update_and_delete_main_record(main_record)

# call the LlaMa function and print the output
print(med.call_LlaMa(api_key="xxxx"))

>> """   
   Neurons are specialized cells that have evolved to process information and transmit it to other cells or
   parts of the body. They have three main parts: the dendrites, the cell body, and the axon. Dendrites are
   branched, tree-like structures that receive signals from other neurons. The cell body, also called the
   soma, contains the nucleus and the rest of the cell's organelles. The axon is a long, thin structure that
   carries signals away from the cell body and to other neurons or to muscles or glands. Neurons are capable of transmitting signals electrically, chemically, or both. They can also store
   information in the form of electrical or chemical changes in their membranes. This information can be retrieved and used later to influence the neuron'
   """

# Load UmlsQA
from umls_qa import UmlsQA

# Initialize the UmlsQA 
med = UmlsQA(model_name="gpt-3.5-turbo", api_key="xxxx")
# Define the medical question in a variable
question = "How does smoking affect lung function?"

# Print the response in English
print(med.ask_medical_question(question))

>> """
   Smoking can significantly impact lung function by causing inflammation 
   and damage to the airways and alveoli. This can lead to conditions such 
   as chronic obstructive pulmonary disease (COPD) and emphysema, which can
   result in difficulty breathing and reduced lung capacity. Smoking also
   increases the risk of developing lung cancer. It is important to quit smoking 
   to protect your lung health and overall well-being.
   """