api-server

src/server/main.py

@@ -1,8 +1,9 @@
 import argparse
+import os
 from typing import List
 from abc import ABC, abstractmethod
 import uvicorn
-from fastapi import FastAPI, File, HTTPException, Query, Request, UploadFile, Depends, Form
+from fastapi import FastAPI, File, HTTPException, Query, Request, UploadFile
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import RedirectResponse, StreamingResponse
 from pydantic import BaseModel, Field
@@ -120,7 +121,7 @@ class TTSService(ABC):
 class ExternalTTSService(TTSService):
     async def generate_speech(self, payload: dict) -> requests.Response:
         try:
-            base_url = "http://example.com/v1/audio/speech"  # Placeholder URL
+            base_url = f"{os.getenv('EXTERNAL_API_BASE_URL')}/v1/audio/speech"
             return requests.post(
                 base_url,
                 json=payload,
@@ -224,7 +225,7 @@ async def chat(
     logger.info(f"Received prompt: {chat_request.prompt}, src_lang: {chat_request.src_lang}")
     
     try:
-        external_url = "http://example.com/v1/chat"  # Placeholder URL
+        external_url = f"{os.getenv('EXTERNAL_API_BASE_URL')}/v1/chat"
         payload = {
             "prompt": chat_request.prompt,
             "src_lang": chat_request.src_lang,
@@ -281,7 +282,7 @@ async def transcribe_audio(
         file_content = await file.read()
         files = {"file": (file.filename, file_content, file.content_type)}
         
-        external_url = f"http://example.com/v1/transcribe/?language={language}"  # Placeholder URL
+        external_url = f"{os.getenv('EXTERNAL_API_BASE_URL')}/v1/transcribe/?language={language}"
         response = requests.post(
             external_url,
             files=files,
@@ -330,7 +331,7 @@ async def translate(
 
     logger.info(f"Received translation request: {len(request.sentences)} sentences, src_lang: {request.src_lang}, tgt_lang: {request.tgt_lang}")
 
-    external_url = "http://example.com/v1/translate"  # Placeholder URL
+    external_url = f"{os.getenv('EXTERNAL_API_BASE_URL')}/v1/translate"
 
     payload = {
         "sentences": request.sentences,
@@ -414,7 +415,7 @@ async def extract_text(
         file_content = await file.read()
         files = {"file": (file.filename, file_content, file.content_type)}
         
-        external_url = f"http://example.com/extract-text/?page_number={page_number}&language={language}"  # Placeholder URL
+        external_url = f"{os.getenv('EXTERNAL_API_BASE_URL')}/extract-text/?page_number={page_number}&language={language}"
         response = requests.post(
             external_url,
             files=files,
@@ -481,7 +482,7 @@ async def visual_query(
         "tgt_lang": visual_query_request.tgt_lang
     })
     
-    external_url = f"http://example.com/v1/visual_query/?src_lang={visual_query_request.src_lang}&tgt_lang={visual_query_request.tgt_lang}"  # Placeholder URL
+    external_url = f"{os.getenv('EXTERNAL_API_BASE_URL')}/v1/visual_query/?src_lang={visual_query_request.src_lang}&tgt_lang={visual_query_request.tgt_lang}"
     
     try:
         file_content = await file.read()
@@ -554,7 +555,7 @@ async def speech_to_speech(
     try:
         file_content = await file.read()
         files = {"file": (file.filename, file_content, file.content_type)}
-        external_url = f"http://example.com/v1/speech_to_speech?language={language}"  # Placeholder URL
+        external_url = f"{os.getenv('EXTERNAL_API_BASE_URL')}/v1/speech_to_speech?language={language}"
 
         response = requests.post(
             external_url,
@@ -585,6 +586,11 @@ async def speech_to_speech(
         raise HTTPException(status_code=500, detail=f"External API error: {str(e)}")
 
 if __name__ == "__main__":
+    # Ensure EXTERNAL_API_BASE_URL is set
+    external_api_base_url = os.getenv("EXTERNAL_API_BASE_URL")
+    if not external_api_base_url:
+        raise ValueError("Environment variable EXTERNAL_API_BASE_URL must be set")
+    
     parser = argparse.ArgumentParser(description="Run the FastAPI server.")
     parser.add_argument("--port", type=int, default=8000, help="Port to run the server on.")
     parser.add_argument("--host", type=str, default="0.0.0.0", help="Host to run the server on.")