commit application

.gitignore

@@ -0,0 +1 @@
+__pycache__/

Dockerfile

@@ -0,0 +1,34 @@
+# syntax=docker/dockerfile:1
+
+FROM python:3.10-slim
+
+ENV PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1 \
+    PIP_NO_CACHE_DIR=1 \
+    HF_HOME=/root/.cache/huggingface \
+    TRANSFORMERS_CACHE=/root/.cache/huggingface/transformers \
+    TOKENIZERS_PARALLELISM=false
+
+WORKDIR /app
+
+# System deps (for scientific libs and uvicorn)
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy only requirements first for better layer caching
+COPY requirements.txt ./
+
+RUN pip install --upgrade pip && \
+    pip install -r requirements.txt
+
+# Copy the rest of the application
+COPY . .
+
+EXPOSE 8000
+
+# Default command: run the main.py runner
+CMD ["python", "main.py"]
+
+

main.py

@@ -0,0 +1,10 @@
+import os
+import uvicorn
+
+
+if __name__ == "__main__":
+    host = os.getenv("HOST", "0.0.0.0")
+    port = int(os.getenv("PORT", "8000"))
+    uvicorn.run("multilingual.multi:app", host=host, port=port, reload=False)
+
+

models/model.py

@@ -0,0 +1,12 @@
+## loading translation model
+
+from transformers import pipeline
+
+model_id = "drrobot9/nllb-ig-yo-ha-finetuned"
+
+translator = pipeline("translation", model=model_id, device=-1)
+
+if __name__ == "__main__":
+    test_text = "Kedu ihe ị na-eme?"
+    translation = translator(test_text, src_lang="ibo_Latn", tgt_lang="eng_Latn")
+    print("Test Translation (Igbo -> English):", translation[0]['translation_text'])

multilingual/multi.py

@@ -0,0 +1,187 @@
+## translation layer for Farmlingua (Milestone2)
+
+import nest_asyncio
+from pyngrok import ngrok
+import uvicorn
+from fastapi import FastAPI
+from pydantic import BaseModel
+import requests
+import os
+from models.model import translator
+
+nest_asyncio.apply()
+
+lang_codes = {
+    "en": "eng_Latn",
+    "ig": "ibo_Latn",
+    "ha": "hau_Latn",
+    "yo": "yor_Latn",
+    "sw": "swh_Latn"
+}
+
+app = FastAPI(
+    title="Farmlingua Translation API",
+    description="Multilingual ability for farmlingua supporting Igbo, Hausa, Yoruba, and Swahili",
+    version="0.0.1"
+)
+
+class Query(BaseModel):
+    text: str
+    lang: str
+
+def translate(text: str, src: str, tgt: str) -> str:
+    try:
+        out = translator(text, src_lang=lang_codes[src], tgt_lang=lang_codes[tgt])
+        return out[0]["translation_text"]
+    except Exception as e:
+        return f"Translation error: {str(e)}"
+
+def query_milestone1(prompt: str) -> str:
+    try:
+        response = requests.post(
+            "https://remostart-milestone-one-farmlingua-ai.hf.space/ask",
+            json={"query": prompt},
+            timeout=200
+        )
+        if response.status_code == 200:
+            data = response.json()
+            return data.get("answer") or str(data)
+        else:
+            return f"Base model error ({response.status_code}): {response.text}"
+    except Exception as e:
+        return f"Failed to connect to Milestone 1: {str(e)}"
+
+# Root endpoint
+@app.get("/")
+def root():
+    """Root endpoint with API information"""
+    return {
+        "message": "Welcome to Farmlingua Translation API",
+        "version": "2.0",
+        "supported_languages": {
+            "igbo": "/igbo",
+            "hausa": "/hausa", 
+            "yoruba": "/yoruba",
+            "swahili": "/swahili",
+            "multilingual": "/ask"
+        },
+        "documentation": "/docs"
+    }
+
+@app.get("/health")
+def health_check():
+    """Health check endpoint"""
+    return {"status": "healthy", "model_loaded": translator is not None}
+
+#  endpoints for each language
+@app.post("/igbo")
+def igbo_endpoint(query: Query):
+    """Endpoint specifically for Igbo language queries"""
+    user_text = query.text
+    
+    # Translate Igbo to English
+    english_text = translate(user_text, "ig", "en")
+    
+    # Get response from Milestone 1
+    english_response = query_milestone1(english_text)
+    
+    # Translate response back to Igbo
+    igbo_response = translate(english_response, "en", "ig")
+    
+    return {
+        "language": "Igbo",
+        "original_text": user_text,
+        "english_translation": english_text,
+        "response": igbo_response
+    }
+
+@app.post("/hausa")
+def hausa_endpoint(query: Query):
+    """Endpoint specifically for Hausa language queries"""
+    user_text = query.text
+    
+    # Translate Hausa to English
+    english_text = translate(user_text, "ha", "en")
+    
+    # Get response from Milestone 1
+    english_response = query_milestone1(english_text)
+    
+    # Translate response back to Hausa
+    hausa_response = translate(english_response, "en", "ha")
+    
+    return {
+        "language": "Hausa",
+        "original_text": user_text,
+        "english_translation": english_text,
+        "response": hausa_response
+    }
+
+@app.post("/yoruba")
+def yoruba_endpoint(query: Query):
+    """Endpoint specifically for Yoruba language queries"""
+    user_text = query.text
+    
+    # Translate Yoruba to English
+    english_text = translate(user_text, "yo", "en")
+    
+    # Get response from Milestone 1
+    english_response = query_milestone1(english_text)
+    
+    # Translate response back to Yoruba
+    yoruba_response = translate(english_response, "en", "yo")
+    
+    return {
+        "language": "Yoruba",
+        "original_text": user_text,
+        "english_translation": english_text,
+        "response": yoruba_response
+    }
+
+@app.post("/swahili")
+def swahili_endpoint(query: Query):
+    """Endpoint specifically for Swahili language queries"""
+    user_text = query.text
+    
+    # Translate Swahili to English
+    english_text = translate(user_text, "sw", "en")
+    
+    # Get response from Milestone 1
+    english_response = query_milestone1(english_text)
+    
+    # Translate response back to Swahili
+    swahili_response = translate(english_response, "en", "sw")
+    
+    return {
+        "language": "Swahili",
+        "original_text": user_text,
+        "english_translation": english_text,
+        "response": swahili_response
+    }
+
+@app.post("/ask")
+def multilingual_ask(query: Query):
+    user_text = query.text
+    user_lang = query.lang.lower()
+
+    if user_lang not in lang_codes:
+        return {"error": "Unsupported language. Use one of: en, ig, ha, yo, sw."}
+
+    if user_lang != "en":
+        english_text = translate(user_text, user_lang, "en")
+    else:
+        english_text = user_text
+
+    english_response = query_milestone1(english_text)
+
+    if user_lang != "en":
+        final_response = translate(english_response, "en", user_lang)
+    else:
+        final_response = english_response
+
+    return {"response": final_response}
+
+# Start ngrok tunnel and run FastAPI server
+if __name__ == "__main__":
+    public_url = ngrok.connect(8000)
+    print(f"Public URL: {public_url}")
+    uvicorn.run(app, host="0.0.0.0", port=8000)

requirements.txt

@@ -0,0 +1,8 @@
+fastapi
+uvicorn
+pydantic
+requests
+transformers
+torch
+nest-asyncio
+pyngrok