Update main.py

main.py

@@ -88,4 +88,25 @@ async def generate_response(data: InputData) -> Dict[str, Any]:
 
     except Exception as e:
         print(f"Model {data.model} failed with error: {e}")
-        raise HTTPException(status_code=500, detail=f"Model {data.model} failed to generate response")
+        raise HTTPException(status_code=500, detail=f"Model {data.model} failed to generate response")
+
+@app.post("/get-medieval-name/")
+async def get_medieval_name() -> Dict[str, str]:
+    try:
+        file_path = "medieval_names.txt"
+        if not os.path.exists(file_path):
+            raise HTTPException(status_code=404, detail="File not found")
+        
+        with open(file_path, "r") as file:
+            names = file.read().splitlines()
+        
+        if not names:
+            raise HTTPException(status_code=404, detail="No names found in the file")
+        
+        random_name = random.choice(names)
+        
+        return {"name": random_name}
+    
+    except Exception as e:
+        print(f"Error: {e}")
+        raise HTTPException(status_code=500, detail="An error occurred while processing the request")