Creating TG Bot

Dockerfile

@@ -0,0 +1,16 @@
+# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["gunicorn", "--workers", "1", "--timeout", "120", "--bind", "0.0.0.0:7860", "app:app"]

app.py

@@ -0,0 +1,235 @@
+import os
+import io
+import torch
+import torch.nn as nn
+from torchvision import models as torchvision_models # Explicitly use torchvision models
+from torchvision import transforms
+from PIL import Image
+from flask import Flask, request, jsonify
+from huggingface_hub import hf_hub_download # To download the specific .pth file
+import os
+import json
+from groq import Groq
+import logging
+from telegram import Update
+from telegram.ext import (
+    ApplicationBuilder,
+    ContextTypes,
+    CommandHandler,
+    MessageHandler,
+    filters
+)
+from google import genai
+from google.genai import types
+
+
+API_KEY_1 = os.getenv("API_KEY")
+API_KEY_2 = os.getenv("GEMINI")
+TASK = os.getenv("TASK")
+client = genai.Client(api_key=API_KEY_2)
+SYSTEM = os.getenv("SYSTEM")
+
+# Optional: Set up logging if required for production.
+logging.basicConfig(level=logging.INFO)
+
+def extract_json_from_response(response):
+    message_body = response.text
+
+    # Try to locate code block with triple backticks
+    if "```" in message_body:
+        start = message_body.find("```")
+        end = message_body.find("```", start + 3)
+        
+        if end != -1:
+            # Strip optional 'json' or other format specifier
+            code_block = message_body[start + 3:end].strip()
+            if code_block.startswith("json"):
+                code_block = code_block[4:].strip()
+            message_body = code_block
+        else:
+            # Only one set of backticks found, possibly malformed
+            message_body = message_body[start + 3:].strip()
+    
+    # If there's no backtick block, assume it's plain JSON
+    try:
+        parsed_json = json.loads(message_body)
+        logging.info("Parsed JSON: %s", parsed_json)
+        return parsed_json
+    except json.JSONDecodeError as e:
+        logging.error("Failed to parse JSON: %s", e)
+        return None
+    
+
+def translate(message):
+    try:
+        # Request to generate content using the model
+        response = client.models.generate_content(
+            model="gemini-2.0-flash",
+            config=types.GenerateContentConfig(system_instruction=SYSTEM),
+            contents=message
+        )
+        
+
+        message_body = extract_json_from_response(response)
+
+        return message_body
+    
+    except Exception as e:
+        logging.error(f"Error while translating message: {e}")
+        return {}  # Return an empty dict in case of any unexpected errors
+
+def gen_message(data_dict):
+    logging.info(data_dict)
+    if not data_dict['symptoms'] or not data_dict['diseases']:
+        message_body = f"""
+    Hi there, I'm Med-AI-Care, your virtual health assistant.
+
+I understood your message as:  
+**"{data_dict['translation']}"**
+
+Thank you for reaching out. Right now, I couldn't detect specific symptoms or conditions based on your message. But you're not alone — it's okay to feel unsure, and I'm still here to support you.
+
+**My Analysis**:  
+{data_dict['analysis']}
+
+**Recommendation**:  
+{data_dict['recommendation']}
+
+I'm an AI assistant, not a doctor. For any health concern, please consult a licensed medical professional.
+
+Take care, and feel free to share more if you'd like me to help further.
+    """.replace("*","\n")
+    else:
+        message_body = f"""
+    Hi there, I'm Med-AI-Care, your virtual health assistant.
+
+You said:  
+**"{data_dict['translation']}"**
+
+Here are the symptoms I picked up:  
+**{', '.join(data_dict['symptoms']).capitalize()}**
+
+Possible conditions based on your symptoms:  
+{chr(10).join([f"- {disease}: {likelihood}" for disease, likelihood in data_dict['diseases'].items()])}
+
+**My Analysis**:  
+{data_dict['analysis']}
+
+📝 **Recommendation**:  
+{data_dict['recommendation']}
+
+I'm just an AI helper — not a doctor. Please follow up with a healthcare professional for proper diagnosis and treatment.
+
+Thanks for reaching out. I'm here if you need more help.
+    """.replace("*","\n")
+        
+        return message_body
+
+def chat(cmd):
+    client = Groq(api_key=API_KEY_1)
+    completion = client.chat.completions.create(
+        model="meta-llama/llama-4-scout-17b-16e-instruct",
+        messages=[
+        {
+            "role": "system",
+            "content": TASK
+        },
+        {
+            "role": "user",
+            "content": cmd
+        }
+        ],
+        temperature=1,
+        max_completion_tokens=1024,
+        top_p=1,
+        stream=True,
+        stop=None,
+    )
+
+    output = []
+    for chunk in completion:
+        output.append(chunk.choices[0].delta.content or "")
+
+
+    in_block = False
+    result = ""
+
+    for line in output:
+        if line.strip() == "```":
+            if not in_block:
+                in_block = True
+            else:
+                break
+        elif in_block and line.strip().lower() != "json":
+            result+=line
+
+
+    try:
+        result_json = json.loads(result)
+        print(result_json)
+        lang = result_json["input_lang"]
+        if lang != "English":
+            data_dict = translate(result)
+            message_body = gen_message(data_dict)
+        else:
+            data_dict = result_json
+            message_body = gen_message(data_dict)
+    except json.JSONDecodeError as e:
+        print("Failed to parse JSON:", e)
+        output,data_dict,message_body = chat(cmd)
+    
+    return output,data_dict,message_body
+
+
+bot_token = "7952675957:AAEQFfIjzTqVdntOsUPRXW62UusPc0zFGLo"
+
+# /start command
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    welcome_message = """
+        Hello! 👋 Welcome to Med-AI-Care, your trusted virtual health assistant.
+
+Our mission is to provide you with intelligent, personalized, and data-driven insights to help you better understand your health. By analyzing the symptoms you share, we aim to offer thoughtful analysis and recommend possible next steps tailored just for you.
+
+Whenever you're ready, just tell me about how you're feeling, and we'll start your health assessment together.
+
+Remember, while I’m here to guide you, this is an AI-powered assistant and not a substitute for professional medical advice. For urgent or serious concerns, please consult a healthcare professional.
+
+Let’s take this step toward better health—together! 💙
+        """
+
+
+    await context.bot.send_message(chat_id=update.effective_chat.id, text = welcome_message)
+
+
+# Handle regular text messages
+async def query(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    message = update.message.text
+    output,data,message_body = chat(message)
+    await context.bot.send_message(chat_id=update.effective_chat.id, text=message_body)
+
+async def error_handler(update: object, context: ContextTypes.DEFAULT_TYPE) -> None:
+    # Log the error or send it to your logging system
+    print(f"Exception while handling an update: {context.error}")
+    
+    # Optional: Notify the user
+    if isinstance(update, Update) and update.effective_chat:
+        await context.bot.send_message(chat_id=update.effective_chat.id,
+                                       text="Server Busy. Please Try Again Later.")
+
+
+# --- Flask App Initialization ---
+app = Flask(__name__)
+
+@app.route("/", methods=["GET"])
+def home():
+
+    return "ConvNeXt Skin Disease Prediction API. Use POST to /predict with an image file."
+
+
+application = ApplicationBuilder().token(bot_token).build()
+
+application.add_handler(CommandHandler('start', start))
+application.add_handler(MessageHandler(filters.TEXT & (~filters.COMMAND), query))
+application.add_error_handler(error_handler)
+application.run_polling()
+

requirements.txt

@@ -0,0 +1,12 @@
+flask
+gunicorn  
+torch
+torchvision
+Pillow
+huggingface_hub
+numpy
+groq
+twilio
+python-telegram-bot
+transformers
+google-genai