Update app.py

app.py

@@ -3,7 +3,8 @@ from typing import List
 
 from fastapi import FastAPI, HTTPException, status
 from pydantic import BaseModel
-import requests  # Import the requests library
+import requests
+from fastapi.responses import HTMLResponse  # Import HTMLResponse
 
 app = FastAPI()
 
@@ -13,6 +14,7 @@ if not GEMINI_API_KEY:
     raise ValueError("The GEMINI_API_KEY environment variable is not set.")
 
 # Gemini API endpoint (using the v1beta endpoint for text generation)
+#  Using gemini-2.0-flash-lite for speed
 GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-lite:generateContent?key=" + GEMINI_API_KEY
 
 
@@ -22,7 +24,7 @@ class TranslationRequest(BaseModel):
     source_language: str = None
 
 
-class TranslationResponse(BaseModel):
+class TranslationResponse(BaseModel):  # Still useful for /detect_language
     translated_text: str
     source_language: str
     target_language: str
@@ -34,7 +36,7 @@ def detect_language_and_options(text: str):
     """Detects the language and provides translation options using the requests library."""
 
     prompt = f"""Please identify the language of the text provided and then offer translation options as numbered choices (1-5).  Use this format:  "The text is in [Language].  Choose a language to translate to: 1. [Option 1], 2. [Option 2], 3. [Option 3], 4. [Option 4], 5. [Option 5]"
-    
+
     Input Text: {text}"""  # Include the input text in the prompt
 
     request_data = {
@@ -68,10 +70,14 @@ def detect_language_and_options(text: str):
         raise HTTPException(status_code=500, detail=f"Error communicating with Gemini API: {e}")
     except Exception as e: #Catch any other unexpected error
         raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {e}")
+
 def translate_with_gemini(text: str, source_language: str, target_language: str) -> str:
-    """Translates text using the Gemini API via requests."""
+    """Translates text, requesting HTML, and handles it correctly."""
+
+    prompt = f"""Translate the following text from {source_language} to {target_language}. Return *only* the translated text.  Format the response as HTML. If any part of the translated text should be emphasized (like an example or key phrase), use `<strong>` tags around that part. Do not include ANY additional text like 'Here is the translation:' or similar.  Only the translated text, formatted as HTML.
 
-    prompt = f"Translate the following text from {source_language} to {target_language}:\n\n{text}"
+    Text:
+    {text}"""
 
     request_data = {
         "contents": [{
@@ -82,26 +88,25 @@ def translate_with_gemini(text: str, source_language: str, target_language: str)
 
     try:
         response = requests.post(GEMINI_API_URL, json=request_data)
-        response.raise_for_status()  # Important: Check for HTTP errors
+        response.raise_for_status()
         response_json = response.json()
 
-        # Robustly extract the translated text, handling potential errors
         try:
             translated_text = response_json['candidates'][0]['content']['parts'][0]['text']
-            return translated_text
+            return translated_text  # Return the raw HTML string
         except (KeyError, IndexError) as e:
             raise HTTPException(status_code=500, detail=f"Error parsing Gemini API response: {e}")
-
     except requests.exceptions.RequestException as e:
         raise HTTPException(status_code=500, detail=f"Error communicating with Gemini API: {e}")
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"An unexpected error occurred {e}")
+        raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {e}")
+
 
 
 
-@app.post("/translate", response_model=TranslationResponse, status_code=status.HTTP_200_OK)
+@app.post("/translate", response_class=HTMLResponse, status_code=status.HTTP_200_OK)
 async def translate(request: TranslationRequest):
-    """Translates text from a source language to a target language."""
+    """Translates text, returning HTML."""
 
     if not request.text:
         raise HTTPException(status_code=400, detail="Text to translate cannot be empty.")
@@ -115,19 +120,19 @@ async def translate(request: TranslationRequest):
             source_language, _ = detect_language_and_options(request.text)
         except HTTPException as e:
             raise e
+    supported_languages = ["Afrikaans","Arabic","Armenian","Azerbaijani","Belarusian","Bosnian","Bulgarian","Catalan","Chinese","Croatian","Czech","Danish","Dutch","English","Estonian","Finnish","French","Galician","German","Greek","Hebrew","Hindi","Hungarian","Icelandic","Indonesian","Italian","Japanese","Kannada","Kazakh","Korean","Latvian","Lithuanian","Macedonian","Malay","Marathi","Maori","Nepali","Norwegian","Persian","Polish","Portuguese","Romanian","Russian","Serbian","Slovak","Slovenian","Spanish","Swahili","Swedish","Tagalog","Tamil","Thai","Turkish","Ukrainian","Urdu","Vietnamese","Welsh",]
 
-    supported_languages = ["English", "Hindi", "Telugu", "Marathi", "Bengali", "Tamil", "Spanish", "French", "German", "Japanese", "Chinese"]
     if request.target_language not in supported_languages:
-        raise HTTPException(status_code=400, detail=f"Target language '{request.target_language}' is not supported.")
+        raise HTTPException(status_code=400, detail=f"Target language '{request.target_language}' is not supported.  Supported languages are: {', '.join(supported_languages)}")
+
 
     try:
-        translated_text = translate_with_gemini(request.text, source_language, request.target_language)
-        return TranslationResponse(translated_text=translated_text, source_language=source_language, target_language=request.target_language)
-    except HTTPException as e:
+        translated_html = translate_with_gemini(request.text, source_language, request.target_language)
+        return translated_html  # FastAPI handles the HTMLResponse correctly
+    except HTTPException as e: # Catch specific HTTP exceptions
         raise e
-    except Exception as e:
-        raise HTTPException(status_code=500,detail=f"An unexpected error occurred {e}")
-
+    except Exception as e: #Catch any other unexpected error
+        raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {e}")
 
 
 @app.post("/detect_language", status_code=status.HTTP_200_OK)