Update app.py

app.py

@@ -1,20 +1,29 @@
 # app.py
 import streamlit as st
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.responses import JSONResponse
+from pydantic import BaseModel
 from typing import List
 import torch
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 from IndicTransToolkit import IndicProcessor
 import json
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.staticfiles import StaticFiles
 import uvicorn
+from starlette.applications import Starlette
+from starlette.routing import Mount, Route
+from starlette.staticfiles import StaticFiles
+import asyncio
+import nest_asyncio
+
+# Enable nested event loops
+nest_asyncio.apply()
 
 # Initialize FastAPI
-api = FastAPI()
+app = FastAPI()
 
 # Add CORS middleware
-api.add_middleware(
+app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
     allow_credentials=True,
@@ -35,6 +44,10 @@ ip = IndicProcessor(inference=True)
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 model = model.to(DEVICE)
 
+class TranslationRequest(BaseModel):
+    sentences: List[str]
+    target_lang: str
+
 def translate_text(sentences: List[str], target_lang: str):
     try:
         src_lang = "eng_Latn"
@@ -78,20 +91,23 @@ def translate_text(sentences: List[str], target_lang: str):
         raise Exception(f"Translation failed: {str(e)}")
 
 # FastAPI routes
-@api.get("/health")
+@app.get("/api/health")
 async def health_check():
     return {"status": "healthy"}
 
-@api.post("/translate")
-async def translate_endpoint(sentences: List[str], target_lang: str):
+@app.post("/api/translate")
+async def translate_endpoint(request: TranslationRequest):
     try:
-        result = translate_text(sentences=sentences, target_lang=target_lang)
-        return result
+        result = translate_text(
+            sentences=request.sentences,
+            target_lang=request.target_lang
+        )
+        return JSONResponse(content=result)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
 # Streamlit interface
-def main():
+def streamlit_app():
     st.title("Indic Language Translator")
     
     # Input text
@@ -133,7 +149,7 @@ def main():
     st.markdown("""
     To use the translation API, send POST requests to:
     ```
-    https://USERNAME-SPACE_NAME.hf.space/translate
+    https://darshankr-trans-en-indic.hf.space/api/translate
     ```
     Request body format:
     ```json
@@ -147,16 +163,16 @@ def main():
     for lang, code in target_languages.items():
         st.markdown(f"- {lang}: `{code}`")
 
+# Create a unified application
+def create_app():
+    routes = [
+        Mount("/api", app),
+        Mount("/", StaticFiles(directory="static", html=True), name="static"),
+    ]
+    return Starlette(routes=routes)
+
 if __name__ == "__main__":
-    # Run both Streamlit and FastAPI
-    import threading
-    
-    def run_fastapi():
-        uvicorn.run(api, host="0.0.0.0", port=8000)
-    
-    # Start FastAPI in a separate thread
-    api_thread = threading.Thread(target=run_fastapi)
-    api_thread.start()
-    
-    # Run Streamlit
-    main()
+    if "streamlit" in sys.argv[0]:
+        streamlit_app()
+    else:
+        uvicorn.run(create_app(), host="0.0.0.0", port=7860)