push

README.md

@@ -1,12 +1,114 @@
----
-title: FARMLINGUAMULTILINGUA
-emoji: 🐢
-colorFrom: yellow
-colorTo: pink
-sdk: docker
-pinned: false
-license: mit
-short_description: multilingual ability of farmlingua
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Farmlingua Multilingual Translation API
+
+A FastAPI-based multilingual translation service supporting Igbo, Hausa, Yoruba, Swahili, and English for agricultural queries.
+
+## Features
+
+- **Simple Language Identifier**: Specify language with `lang` parameter
+- **Single Endpoint**: `/ask` - Send text with language identifier
+- **Supported Languages**: English (en), Igbo (ig), Hausa (ha), Yoruba (yo), Swahili (sw)
+- **Agricultural Focus**: Specialized for farming and agricultural queries
+- **External Integration**: Connects to Milestone 1 Q&A service
+
+## Quick Start
+
+### Local Development
+
+1. **Install Dependencies**
+   ```bash
+   pip install -r requirements.txt
+   ```
+
+2. **Run the Server**
+   ```bash
+   python main.py
+   ```
+
+3. **Test the API**
+   ```bash
+   python test_simple.py
+   ```
+
+4. **Access Documentation**
+   - Interactive API docs: http://localhost:8000/docs
+   - Health check: http://localhost:8000/health
+
+### Docker Deployment
+
+1. **Build Image**
+   ```bash
+   docker build -t farmlingua-api .
+   ```
+
+2. **Run Container**
+   ```bash
+   docker run -p 8000:8000 farmlingua-api
+   ```
+
+## API Usage
+
+### Single Endpoint: `/ask`
+
+**POST** `/ask`
+
+**Request Body:**
+```json
+{
+  "text": "Kedu ka m ga-esi na-akụ ọka?",
+  "lang": "ig"
+}
+```
+
+**Response:**
+```json
+{
+  "language": "Igbo",
+  "language_code": "ig",
+  "original_text": "Kedu ka m ga-esi na-akụ ọka?",
+  "english_translation": "How can I plant grain?",
+  "response": "Jiri oghere mkpụrụ osisi ma ọ bụ osisi aka..."
+}
+```
+
+### Supported Language Codes
+
+| Code | Language | NLLB Code |
+|------|----------|-----------|
+| `en` | English  | `eng_Latn` |
+| `ig` | Igbo     | `ibo_Latn` |
+| `ha` | Hausa    | `hau_Latn` |
+| `yo` | Yoruba   | `yor_Latn` |
+| `sw` | Swahili  | `swh_Latn` |
+
+## Project Structure
+
+```
+FARMLINGUAMULTILINGUA/
+├── models/
+│   └── model.py          # Hugging Face translation model
+├── multilingual/
+│   └── multi.py          # FastAPI application
+├── main.py               # Application entry point
+├── test_simple.py        # Test suite
+├── requirements.txt      # Python dependencies
+├── Dockerfile           # Docker configuration
+├── .gitignore          # Git ignore rules
+└── README.md           # This file
+```
+
+## Environment Variables
+
+- `HOST`: Server host (default: `0.0.0.0`)
+- `PORT`: Server port (default: `8000`)
+
+## Technology Stack
+
+- **FastAPI**: Web framework
+- **Transformers**: Hugging Face model pipeline
+- **NLLB Model**: `drrobot9/nllb-ig-yo-ha-finetuned`
+- **Uvicorn**: ASGI server
+- **Docker**: Containerization
+
+## License
+
+MIT License

main.py

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

multilingual/multi.py

@@ -9,6 +9,7 @@ import requests
 import os
 from models.model import translator
 
+
 nest_asyncio.apply()
 
 lang_codes = {
@@ -19,6 +20,8 @@ lang_codes = {
     "sw": "swh_Latn"
 }
 
+# Simple language identifier - no auto-detection needed
+
 app = FastAPI(
     title="Farmlingua Translation API",
     description="Multilingual ability for farmlingua supporting Igbo, Hausa, Yoruba, and Swahili",
@@ -27,7 +30,7 @@ app = FastAPI(
 
 class Query(BaseModel):
     text: str
-    lang: str
+    lang: str  
 
 def translate(text: str, src: str, tgt: str) -> str:
     try:
@@ -57,13 +60,18 @@ def root():
     """Root endpoint with API information"""
     return {
         "message": "Welcome to Farmlingua Translation API",
-        "version": "2.0",
+        "version": "3.0",
+        "features": {
+            "simple_language_identifier": "Specify language with 'lang' parameter",
+            "single_endpoint": "/ask - Send text with language identifier",
+            "supported_codes": "en, ig, ha, yo, sw"
+        },
         "supported_languages": {
-            "igbo": "/igbo",
-            "hausa": "/hausa", 
-            "yoruba": "/yoruba",
-            "swahili": "/swahili",
-            "multilingual": "/ask"
+            "en": "English",
+            "ig": "Igbo", 
+            "ha": "Hausa",
+            "yo": "Yoruba",
+            "sw": "Swahili"
         },
         "documentation": "/docs"
     }
@@ -73,112 +81,47 @@ 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):
+    """Multilingual endpoint with simple language identifier"""
     user_text = query.text
     user_lang = query.lang.lower()
-
+    
+    # Validate language code
     if user_lang not in lang_codes:
         return {"error": "Unsupported language. Use one of: en, ig, ha, yo, sw."}
-
+    
+    # Get language name for response
+    lang_names = {
+        "en": "English",
+        "ig": "Igbo", 
+        "ha": "Hausa",
+        "yo": "Yoruba",
+        "sw": "Swahili"
+    }
+    
+    # Translate to English if not English
     if user_lang != "en":
         english_text = translate(user_text, user_lang, "en")
     else:
         english_text = user_text
-
+    
+    # Get response from Milestone 1
     english_response = query_milestone1(english_text)
-
+    
+    # Translate response back to original language if not English
     if user_lang != "en":
         final_response = translate(english_response, "en", user_lang)
     else:
         final_response = english_response
-
-    return {"response": final_response}
+    
+    return {
+        "language": lang_names[user_lang],
+        "language_code": user_lang,
+        "original_text": user_text,
+        "english_translation": english_text,
+        "response": final_response
+    }
 
 # Start ngrok tunnel and run FastAPI server
 if __name__ == "__main__":

requirements.txt

@@ -5,4 +5,4 @@ requests
 transformers
 torch
 nest-asyncio
-pyngrok
+pyngrok