Upload 2 files

app.py

@@ -0,0 +1,224 @@
+import gradio as gr
+import requests # To make API calls to Gemini
+import base64
+import os
+from PIL import Image
+import io
+
+# --- Configuration ---
+# IMPORTANT: You must set your GEMINI_API_KEY in your Hugging Face Space secrets.
+# The key should be named GEMINI_API_KEY.
+API_KEY = os.getenv("GEMINI_API_KEY", "")
+GEMINI_API_URL = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent?key={API_KEY}"
+
+# --- Disclaimer (Crucial!) ---
+MANDATORY_DISCLAIMER_TEXT = """
+**⚠️ Important Disclaimer:** This AI tool provides general observations or information based on the input provided.
+It is **NOT** a medical diagnosis and should **NOT** be used to self-diagnose or make treatment decisions.
+The information provided should **NOT** be considered a substitute for professional medical advice, diagnosis, or treatment from a qualified healthcare provider.
+**Always seek the advice of a doctor or other qualified healthcare provider** with any questions you may have regarding a medical condition, symptoms, or health concerns.
+Never disregard professional medical advice or delay in seeking it because of something you have read or seen from this tool.
+If you believe you may have a medical emergency, call your doctor or emergency services immediately.
+"""
+
+# --- AI Prompt for Image Analysis ---
+def get_gemini_image_prompt(user_comment=None):
+    prompt_core = """You are an AI assistant. Your role is to observe visual signs in an image of a human or animal body part and provide general information.
+You are not a medical professional and cannot provide a diagnosis. Your response MUST be general and cautious.
+
+Analyze the following image. Structure your response as follows:
+
+1.  **Observations:** Based *only* on what is visually apparent in the image, describe any notable features, marks, discolorations, or unusual appearances. Use neutral, descriptive language. Do not infer or guess beyond what is visible. If the image is unclear or no specific observations can be made, state that.
+
+2.  **General Information (Not a Diagnosis):** Briefly mention what such visual signs *could generally be associated with* in common, non-emergency situations. Avoid definitive statements or listing specific conditions. For example, instead of saying 'This is X condition,' say 'Sometimes, redness can be associated with irritation.' If no specific observations were made, state that general well-being is important.
+
+3.  **General Prevention/Wellness Tips:** Offer *broad* wellness or preventative tips related to the depicted body part or general health. These tips should be generic and not specific to any presumed condition. For example, 'Maintaining good hygiene is important for skin health,' or 'A balanced diet supports overall well-being.'
+
+4.  **Mandatory Disclaimer:** Your entire response MUST begin with the following disclaimer, exactly as written, with no text preceding it:
+
+""" + MANDATORY_DISCLAIMER_TEXT
+
+    user_guidance_text = ""
+    if user_comment and user_comment.strip():
+        user_guidance_text += f"\n\nThe user has provided the following comment, possibly related to an area they have highlighted or are concerned about: \"{user_comment.strip()}\". Please take this comment into consideration during your observation. If the user has drawn on the image, pay particular attention to the annotated areas in conjunction with their comment."
+    else:
+        user_guidance_text += "\n\nThe user may have drawn on the image to highlight an area of interest. If so, please pay particular attention to any such visual annotations during your observation."
+
+    prompt_suffix = """
+
+Ensure your language is cautious, non-prescriptive, and strongly emphasizes the need to consult a real medical professional throughout your response if appropriate (e.g., by reiterating "consult a doctor for any concerns").
+If the image is not of a body part or is inappropriate, state that you can only analyze images of body parts for general observations and include the disclaimer.
+"""
+    return prompt_core + user_guidance_text + prompt_suffix
+
+# --- AI Prompt for Text Queries ---
+def get_gemini_text_prompt(user_query):
+    return f"""You are an AI assistant designed to provide general health-related information based on user queries.
+You are NOT a medical professional and CANNOT provide a diagnosis or medical advice. Your response MUST be general, informative, and cautious.
+
+The user has asked the following question: "{user_query}"
+
+Please structure your response as follows:
+
+1.  **General Information:** Based on widely accepted health knowledge, provide a general answer to the user's query. Avoid definitive statements or specific medical advice. If the question is too specific, ambiguous, or outside the scope of general health information, state that and recommend consulting a professional.
+
+2.  **General Prevention/Wellness Tips:** If applicable to the query, offer *broad* wellness or preventative tips related to the topic. These tips should be generic. For example, if the query is about headaches, you might mention 'Staying hydrated and managing stress can be helpful for overall well-being.'
+
+3.  **When to See a Doctor:** Briefly mention general circumstances under which one should consult a healthcare professional regarding the topic of the query. For example, 'If symptoms are severe, persistent, or worsening, it's important to see a doctor.'
+
+4.  **Mandatory Disclaimer:** Your entire response MUST begin with the following disclaimer, exactly as written, with no text preceding it:
+
+""" + MANDATORY_DISCLAIMER_TEXT + """
+
+Ensure your language is cautious and strongly emphasizes that the information is not a substitute for professional medical advice.
+If the query is inappropriate, not health-related, or seeks specific medical diagnosis/treatment, politely decline to answer directly and instead reiterate the importance of consulting a qualified healthcare provider, including the full disclaimer.
+"""
+
+# --- Image Processing and API Call ---
+def analyze_image_with_ai(pil_image, user_comment): # Added user_comment
+    if not API_KEY:
+        return MANDATORY_DISCLAIMER_TEXT + "\n\n**Error: `GEMINI_API_KEY` is not set. Please configure it in your Hugging Face Space secrets.**"
+    if pil_image is None:
+        # Gradio's Image component with tool='paint' might return None if no image is drawn or uploaded initially.
+        # Or if the input is cleared.
+        return MANDATORY_DISCLAIMER_TEXT + "\n\n**Error: No image uploaded or drawn. Please provide an image.**"
+
+    try:
+        # The pil_image from Gradio (with tool='paint') will include any drawings.
+        buffered = io.BytesIO()
+        if pil_image.mode == 'RGBA' or pil_image.mode == 'P':
+            pil_image = pil_image.convert('RGB')
+        pil_image.save(buffered, format="JPEG")
+        img_byte = buffered.getvalue()
+        img_base64 = base64.b64encode(img_byte).decode("utf-8")
+    except Exception as e:
+        print(f"Error converting image: {e}")
+        return MANDATORY_DISCLAIMER_TEXT + f"\n\n**Error: Could not process the uploaded image. Details: {e}**"
+
+    prompt_text = get_gemini_image_prompt(user_comment) # Pass user_comment
+    payload = {
+        "contents": [
+            {
+                "role": "user",
+                "parts": [
+                    {"text": prompt_text},
+                    {"inlineData": {"mimeType": "image/jpeg", "data": img_base64}}
+                ]
+            }
+        ],
+        "generationConfig": {"temperature": 0.4, "topK": 32, "topP": 1, "maxOutputTokens": 2000}, # Slightly increased tokens
+        "safetySettings": [
+            {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+            {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+            {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+            {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+        ]
+    }
+    return make_gemini_api_call(payload)
+
+# --- Text Query Processing and API Call ---
+def answer_text_query(text_query):
+    if not API_KEY:
+        return MANDATORY_DISCLAIMER_TEXT + "\n\n**Error: `GEMINI_API_KEY` is not set. Please configure it in your Hugging Face Space secrets.**"
+    if not text_query or text_query.strip() == "":
+        return MANDATORY_DISCLAIMER_TEXT + "\n\n**Error: No question provided. Please enter your health-related question.**"
+
+    prompt_text = get_gemini_text_prompt(text_query)
+    payload = {
+        "contents": [{"role": "user", "parts": [{"text": prompt_text}]}],
+        "generationConfig": {"temperature": 0.7, "topK": 40, "topP": 1, "maxOutputTokens": 1500},
+        "safetySettings": [
+            {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+            {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+            {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+            {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+        ]
+    }
+    return make_gemini_api_call(payload)
+
+# --- Common function to make API call and handle response ---
+def make_gemini_api_call(payload):
+    try:
+        response = requests.post(GEMINI_API_URL, json=payload, headers={'Content-Type': 'application/json'})
+        response.raise_for_status()
+        result = response.json()
+
+        if (result.get("candidates") and
+            len(result["candidates"]) > 0 and
+            result["candidates"][0].get("content") and
+            result["candidates"][0]["content"].get("parts") and
+            len(result["candidates"][0]["content"]["parts"]) > 0 and
+            result["candidates"][0]["content"]["parts"][0].get("text")):
+            generated_text = result["candidates"][0]["content"]["parts"][0]["text"]
+            if not generated_text.strip().startswith("**⚠️ Important Disclaimer:**"):
+                return MANDATORY_DISCLAIMER_TEXT + "\n\n" + generated_text.lstrip()
+            return generated_text
+        elif result.get("promptFeedback") and result["promptFeedback"].get("blockReason"):
+            block_reason = result['promptFeedback']['blockReason']
+            safety_ratings = result['promptFeedback'].get('safetyRatings', 'No specific safety ratings provided.')
+            return MANDATORY_DISCLAIMER_TEXT + f"\n\n**Error from AI: The request was blocked due to: {block_reason}.**\nSafety details: {safety_ratings}"
+        else:
+            print(f"Unexpected API response structure: {result}")
+            return MANDATORY_DISCLAIMER_TEXT + "\n\n**Error: Could not retrieve a valid response from the AI model.**"
+    except requests.exceptions.RequestException as e:
+        print(f"Request failed: {e}")
+        return MANDATORY_DISCLAIMER_TEXT + f"\n\n**Error: Failed to connect to the AI model. Details: {e}**"
+    except Exception as e:
+        print(f"An unexpected error occurred: {e}")
+        return MANDATORY_DISCLAIMER_TEXT + f"\n\n**Error: An unexpected error occurred. Details: {e}**"
+
+# --- Gradio Interface ---
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="orange")) as demo:
+    gr.Markdown(
+        "# 🩺 AI Health Information Assistant (Experimental)"
+        "\nUse the tabs below to either analyze an image (with optional drawing and comments) or ask a general health-related question."
+    )
+    gr.Markdown(MANDATORY_DISCLAIMER_TEXT, elem_id="static-disclaimer")
+
+    with gr.Tabs():
+        with gr.TabItem("🖼️ Image Analyzer"):
+            gr.Markdown("Upload an image of a human or animal body part. You can draw on the image to highlight areas and add comments. The AI will attempt to describe observable signs and offer *general wellness information only*.")
+            with gr.Row():
+                with gr.Column(scale=1):
+                    # Image input with drawing tool enabled - REMOVED tool="paint"
+                    image_input = gr.Image(type="pil", label="Upload Image & Draw (Optional)", sources=["upload", "webcam", "clipboard"], interactive=True)
+                    # Textbox for user comments on the image
+                    user_comment_image = gr.Textbox(lines=3, placeholder="Add any comments about the image or highlighted areas here (optional)...", label="Your Comments on Image")
+                    image_submit_button = gr.Button("Analyze Image", variant="primary")
+                with gr.Column(scale=2):
+                    image_output_text = gr.Markdown(label="AI Observations & General Guidance")
+            
+            # Update click function to pass both image and comment
+            image_submit_button.click(analyze_image_with_ai, inputs=[image_input, user_comment_image], outputs=image_output_text)
+            
+            gr.Examples(
+                examples=[], # Add example image paths if you wish, but drawing needs user interaction
+                inputs=[image_input, user_comment_image], # inputs for examples
+                cache_examples=False,
+                label="Example Scenarios (Load image, then optionally draw & comment)"
+            )
+
+        with gr.TabItem("❓ Text Query"):
+            gr.Markdown("Ask a general health-related question. The AI will provide *general information and wellness tips only*.")
+            with gr.Row():
+                with gr.Column(scale=1):
+                    text_input = gr.Textbox(lines=5, placeholder="Type your health-related question here...\ne.g., What are some general tips for better sleep?", label="Your Question")
+                    text_submit_button = gr.Button("Get Information", variant="primary")
+                with gr.Column(scale=2):
+                    text_output_text = gr.Markdown(label="AI Response & General Guidance")
+            text_submit_button.click(answer_text_query, inputs=text_input, outputs=text_output_text)
+            gr.Examples(
+                examples=[
+                    ["What are some general tips for managing stress?"],
+                    ["Can you give some general advice for maintaining a healthy diet?"],
+                    ["What are common ways to improve sleep hygiene?"],
+                ],
+                inputs=text_input,
+                cache_examples=False,
+                label="Example Questions (click to load)"
+            )
+
+    gr.Markdown("--- \n *This tool is for informational purposes only and does not provide medical advice. Always consult a qualified healthcare professional for any health concerns or before making any decisions related to your health.*")
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+requests
+Pillow