Update app.py

app.py

@@ -1,76 +1,103 @@
-"""
 import os
-from ibm_watson_machine_learning.foundation_models import Model
-from ibm_watson_machine_learning.metanames import GenTextParamsMetaNames as GenParams
-from ibm_watson_machine_learning.foundation_models.utils.enums import ModelTypes, DecodingMethods
 import gradio as gr
+from huggingface_hub import InferenceClient
 
+# Set up the API key and model for Deepseek
+DEEPSEEK_API_KEY = os.getenv("Deepseek_API_Key")
+DEEPSEEK_MODEL = "deepseek-chat"  # Replace with the correct model name if needed
 
-# Set up the API key and project ID for IBM Watson
-watsonx_API = os.environ.get("watsonx_API") 
-project_id = os.environ.get("project_id")
+# Initialize the Deepseek Inference client
+client = InferenceClient(model=DEEPSEEK_MODEL, token=DEEPSEEK_API_KEY)
 
-# Generation parameters
-gen_parms = {
-    "max_new_tokens": 512,  # Adjust as needed 
-    "temperature": 0.7  # Adjust for creativity
-}
 
-# Model and project settings
-model_id = "meta-llama/llama-2-13b-chat"
+def generate_game(auditory_memory, selective_attention, dichotic_listening, phonemic_awareness, temporal_processing, language_comprehension):
+    # Craft the prompt for the model
+    prompt = f"""You are an experienced learning disabilities therapist AI. 
+Provide customized advice to choose the right remediation game suitable for a Central Auditory Processing Disorder 'CAPD' patient
+(user of the chatbot could be the patient, a family member, a teacher or a therapist helping the patient). 
+- Problems related to auditory memory: {auditory_memory}
+- Problems related to selective attention: {selective_attention}
+- Problems related to dichotic listening: {dichotic_listening}
+- Problems related to phonemic awareness: {phonemic_awareness}
+- Problems related to temporal processing: {temporal_processing}
+- Problems related to language comprehension: {language_comprehension}
 
-credentials={
-        "apikey": watsonx_API,
-        "url": "https://us-south.ml.cloud.ibm.com"
-    }
+Include recommendations from the following:
+- Recommended Games for Auditory Memory:
+    "Memory Stories", "Memory Sounds", "Memory Phoneme, Words, Digits", "Memory Digits", "Sustained Attention"
+- Recommended Games for Selective Attention:
+    "Selective Auditory Attention (Sentences)", "Sustained Attention", "Numbers and Letters"
+- Recommended Games for Dichotic Listening:
+    "Dichotic Listening (Letter - Words - Digits)", "Dichotic Listening (Sentences)"
+- Recommended Games for Phonemic Awareness:
+    "Blending", "Rhyme Detection", "Segmentation", "Vowel Detection", "Missing Parts"
+- Recommended Games for Temporal Processing:  
+    "Duration Discrimination", "Frequency Sequencing", "Frequency Discrimination", "Temporal Processing (Phoneme Identification)", "Gap Detection", 
+    "Duration Sequencing", "Temporal Processing - Phonic Match", "Block Commander"
+- Recommended Games for Language Comprehension:
+    "Language Comprehension", "Memory Stories"
+Be concise and limit your response to 512 tokens or less."""
 
-model = Model(
-    model_id = 'meta-llama/llama-2-13b-chat', # you can also specify like: ModelTypes.LLAMA_2_70B_CHAT
-    params = gen_parms,
-    credentials={
-        "apikey": watsonx_API,
-        "url": "https://us-south.ml.cloud.ibm.com"
-    },
-    project_id= project_id
-    )
+    try:
+        # Call the Deepseek API
+        response = client.chat_completion(
+            messages=[
+                {"role": "system", "content": "You are a helpful and knowledgeable learning disabilities expert with knowledge of Central Auditory Processing Disorder (CAPD)."},
+                {"role": "user", "content": prompt},
+            ],
+            max_tokens=512,
+            temperature=0.7,
+        )
+        # Extract the response text
+        game_advice = response.choices[0].message.content.strip()
+    except Exception as e:
+        game_advice = f"An error occurred: {str(e)}"
 
+    return game_advice
 
-# Initialize the model
-model = Model(model_id, credentials, gen_parms, project_id)
 
-# Function to generate customized career advice
-def generate_career_advice(field, position_name, current_qualifications, likes, skills):
-    # Craft the prompt for the model 
-    prompt = f"Generate a customized career advice using desired career field: {field}, \
-    dream job: {position_name}, \
-    current qualifications: {current_qualifications}, \
-    likes: {likes}, \
-    skills: {skills}."
-
-    generated_response = model.generate(prompt, gen_parms)
-
-    # Extract the generated text
-    career_advice = generated_response["results"][0]["generated_text"]
-    return career_advice
-
-
-# Create Gradio interface for the cover letter generation application
-career_advice_app = gr.Interface(
-    fn=generate_career_advice,
-    allow_flagging="never", # Deactivate the flag function in gradio as it is not needed.
+# Create Gradio interface for the CAPD Game application
+CAPD_Game_app = gr.Interface(
+    fn=generate_game,
+    allow_flagging="never",
     inputs=[
-        gr.Textbox(label="Desired Career Field (e.g., healthcare, trades, social service, etc., or enter 'not sure')", placeholder="Enter the field which you are interested in... or type 'not sure'."),
-        gr.Textbox(label="Your Dream Job (e.g., nurse, personal support worker, software developer, plumber, etc., or enter 'not sure')", placeholder="Enter the name of the position you are interested in... or type 'not sure'"),
-        gr.Textbox(label="Current Qualifications (e.g., studying in high school, high school diploma, college diploma, etc.)", placeholder="Enter your current qualifications ..."),
-        gr.Textbox(label="Likes (e.g., I like working with my hands, I like to work outside, I like to help people, I like teaching, ...)", placeholder="Enter activities you like ...", lines=10),
-        gr.Textbox(label="Skills (e.g., I am good at math, science, languages, computers, research, hand tools, etc.)", placeholder="Enter your skills ...", lines=10),
+        gr.Textbox(
+            label="Auditory Memory",
+            placeholder="Describe if you have issues with your auditory memory. For instance, if you hear a number of 3 digits, do you tend to forget it? Or if someone mentions a name, do you fail to remember the name?",
+            lines=2,
+        ),
+        gr.Textbox(
+            label="Selective Attention",
+            placeholder="Describe if you experience problems with selective attention. For instance, if there is noise in the room, do you find it hard to understand what is said?",
+            lines=2,
+        ),
+        gr.Textbox(
+            label="Dichotic Listening",
+            placeholder="Describe if you experience problems with listening from both or either ears. For instance, do you have difficulty understanding speech presented to both ears, or difficulty ignoring one message while focusing on another?",
+            lines=2,
+        ),
+        gr.Textbox(
+            label="Phonemic Awareness",
+            placeholder="Describe if you experience difficulty understanding speech, following directions, or misidentifying certain sounds, letters, or words.",
+            lines=2,
+        ),
+        gr.Textbox(
+            label="Temporal Processing",
+            placeholder="Describe if you experience difficulty understanding rapid or distorted speech, or identifying individual words when the gap between words is small.",
+            lines=2,
+        ),
+        gr.Textbox(
+            label="Language Comprehension",
+            placeholder="Describe if you experience difficulty understanding speech, directions, or stories.",
+            lines=2,
+        ),
     ],
-    outputs=gr.Textbox(label="Customized Career Advice"),
-    title="Customized AI-Powered Career Advice - by Wael Nawara",
-    description="This App will generate an AI-powered customized career advice based on the career field which you select, your dream job, current qualifications, likes and skills. A word of caution: even AI makes mistakes!"
+    outputs=gr.Textbox(label="Recommended Games"),
+    title="CAPD Games",
+    description="This App provides general Ai advice regarding CAPD Games. For best results, visit your audiologist or doctor for proper diagnosis and treatment. Disclaimer: AI can make mistakes. Use with caution and at your own risk!",
 )
 
-# Launch the application
-career_advice_app.launch(server_name="0.0.0.0", debug=True, server_port=7860, share=True)
 
-"""
+# Launch the application
+if __name__ == "__main__":
+    CAPD_Game_app.launch(server_name="0.0.0.0", debug=True, share=True)